下面将将tensor转成numpy的几种情况
- GPU中的Variable变量:
a.cuda().data.cpu().numpy()
- GPU中的tensor变量:
a.cuda().cpu().numpy()
-
CPU中的Variable变量:
a.data.numpy() -
CPU中的tensor变量:
a.numpy()
总结:
.cuda()是读取GPU中的数据
.data是读取Variable中的tensor
.cpu是把数据转移到cpu上
.numpy()把tensor变成numpy
转自:[https://www.jianshu.com/p/e9074a6f408d]