pytorch:.cuda() & .cpu() & .data & .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