pytorch和numpy数据之间的转换

（来源于莫烦的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]]