Turbot-DL入门教程篇-Pytorch应用-Tensor和Numpy之间的转换

说明：

• 介绍Pytorch中，Tensor和Numpy之间的转换

环境：

• Python 3.5.2

步骤：

• 新建文件

$ vim pytorch_test.py

• 内容如下

import torch
import numpy as np

np_data = np.arange(6).reshape((2, 3))
torch_data = torch.from_numpy(np_data)
tensor2array = torch_data.numpy()
print(
    '\nnumpy', np_data,
    '\ntorch', torch_data,
    '\ntensor2array', tensor2array
)

• 运行

$ python3 pytorch_test.py

• 结果如下

numpy [[0 1 2]
 [3 4 5]] 
torch tensor([[0, 1, 2],
        [3, 4, 5]], dtype=torch.int32) 
tensor2array [[0 1 2]
 [3 4 5]]