1.numpy转为tensor
np2tensor = torch.fromnumpy(numpy1) # numpy1为ndarray
2.tensor转为numpy
tensor2np = tensor1.numpy() # tensor1为tensor
但是有可能会有这个错误:
TypeError: can’t convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor
那么将tensor先拷贝到CPU再转
tensor2np = tensor1.cpu().numpy() # tensor1为tensor