Pytorch 中遇到can't convert CUDA tensor to numpy问题
假设tensor为 t
1、将tensor转为cpu格式
t=t.cpu()
2、进行numpy的相应操作
3、将数据转为tensor,并转为gpu格式
t=torch.from_numpy(t).cuda()
Pytorch 中遇到can't convert CUDA tensor to numpy问题
假设tensor为 t
1、将tensor转为cpu格式
t=t.cpu()
2、进行numpy的相应操作
3、将数据转为tensor,并转为gpu格式
t=torch.from_numpy(t).cuda()