判断数据是否在gpu

tensor.is_cuda

True / False

posted @ 2022-03-17 10:42  ArdenWang  阅读(110)  评论(0)  编辑  收藏  举报