pytorch学习-pytorch和numpy中数组操作

PyTorch 是 Torch 在 Python 上的衍生.
因为 PyTorch 是一个使用 Python语言的神经网络库, Torch 很好用, 但是 Lua 又不是特别流行, 所有开发团队将 Lua 的 Torch 移植到了更流行的语言 Python 上.
Torch 自称为神经网络界的 Numpy, 因为他能将 torch 产生的 tensor 放在 GPU 中加速运算 (前提是你有合适的 GPU), 就像 Numpy 会把 array 放在 CPU 中加速运算.

#实现pytorch和numpy中数组的转化
import numpy as np
import torch

np_data=np.arange(6).reshape((2,3))
torch_data=torch.from_numpy(np_data)
tensor2array=torch_data.numpy()

print(np_data)
print(torch_data)
print(tensor2array)

输出结果：

[[0 1 2]
 [3 4 5]]
tensor([[0, 1, 2],
        [3, 4, 5]])