判断数据是否在gpu

tensor.is_cuda

True / False

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