Numpy和PyTorch基础知识（个人笔记正在更新）

1.Torch和Numpy之间转换：

In [1]: import torch

In [2]: import numpy as np

In [3]: np_data=np.arange(6).reshape(2,3)

In [4]: tensor_data=torch.from_numpy(np_data)

In [5]: print np_data
[[0 1 2]
 [3 4 5]]

In [6]: print tensor_data
tensor([[0, 1, 2],
        [3, 4, 5]])

In [7]: tensor2array=tensor_data.numpy()

In [8]: print tensor2array
[[0 1 2]
 [3 4 5]]

 

2.Torch的运算举例：

In [9]: data=[-1,-2,1,2]

In [10]: tensor=torch.FloatTensor(data)

In [11]: print torch.abs(tensor)
tensor([1., 2., 1., 2.])

In [12]: np_data=np.array(data)

In [13]: print np.abs(np_data)
[1 2 1 2]

 

3.Variable变量：

如果把tensor数据看作鸡蛋的话，那么variable变量相当于放鸡蛋的篮子，tensor数据不能直接进行反向传播，而variable变量可以反向传播。

In [1]: import torc