Pytorch学习（一）

莫烦python视频学习笔记 视频链接https://www.bilibili.com/video/BV1Vx411j7kT?from=search&seid=3065687802317837578

1、Numpy与Pytorch的数据转换

import torch
import numpy as np

np_data = np.arange(6).reshape((2,3))
torch_data = torch.from_numpy(np_data)      # numpy数据转换为torch数据
tensor2array = torch_data.numpy()           # 再转换成numpy数据
print('\nnumpy:', np_data)
print('\ntorch:', torch_data)
print('\n', tensor2array)

输出：

numpy: [[0 1 2]
 [3 4 5]]
 
torch: tensor([[0, 1, 2],
        [3, 4