(来源于莫烦的pytorch教程)
首先是导入需要的包:
import torch
import numpy as np
从np转到pytorch的tensor:
np_data=np.arange(6).reshape((2,3))
torch_data=torch.from_numpy(np_data)
从tensor转到np
tensor2np=torch_data.numpy()
以下是三个变量的值
np_data [[0 1 2]
[3 4 5]]
torch_data ([[0, 1, 2],
[3, 4, 5]], dtype=torch.int32)
tensor2np [[0 1 2]
[3 4 5]]