下面将将tensor转成numpy的几种情况
1. GPU中的Variable变量:
a.cuda().data.cpu().numpy()
2. GPU中的tensor变量:
a.cuda().cpu().numpy()
3. CPU中的Variable变量:
a.data.numpy()
4. CPU中的tensor变量:
a.numpy()
总结:
.cuda()是读取GPU中的数据
.data是读取Variable中的tensor
.cpu是把数据转移到cpu上
.numpy()把tensor变成numpy
网友评论