pytorch：.cuda() & .cpu() & .data
2020-08-16 本文已影响0人
不太聪明的亚子
下面将将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
