Torch,Numpy,Pandas转换具体实例

1.pandas.dataframe转成torch.tensor
方法一：先转成numpy array，再转成torch tensor

my_array = np.array(my_df)
my_tensor = torch.tensor(my_array)

方法二：若pandas.DataFrame的数据格式统一

torch.from_numpy(pd_data.values)

2.Tensor 和 NumPy 相互转换
可用numpy()和from_numpy()将Tensor和NumPy中的数组相互转换。
但是需要注意的一点是： 这两个函数所产生的Tensor和NumPy中的数组共享相同的内存，所以它们之间的转换很快，改变其中一个时另一个也会改变！
2.1 Tensor 转 NumPy

a = torch.ones(6)
b = a.numpy()
print(a, b)

a += 1
print(a, b)
b += 1
print(a, b)

output：

tensor([1., 1., 1., 1., 1., 1.]) [1. 1. 1. 1. 1. 1.]
tensor([2., 2., 2., 2., 2., 2.]) [2. 2. 2. 2. 2. 2.]
tensor([3., 3., 3., 3., 3., 3.]) [3. 3. 3. 3. 3. 3.]

2.2 NumPy 数组转 Tensor

import numpy as np
a = np.ones(7)
b = torch.from_numpy(a)
print(a, b)

a += 1
print(a, b)
b += 1
print(a, b)

output：

[1. 1. 1. 1. 1. 1. 1.] tensor([1., 1., 1., 1., 1., 1., 1.], dtype=torch.float64)
[2. 2. 2. 2. 2. 2. 2.] tensor([2., 2., 2., 2., 2., 2., 2.], dtype=torch.float64)
[3. 3. 3. 3. 3. 3. 3.] tensor([3., 3., 3., 3., 3., 3., 3.], dtype=torch.float64)

3.3 torch.tensor() 将 NumPy 数组转换成 Tensor
直接用torch.tensor()将NumPy数组转换成Tensor，该方法总是会进行数据拷贝，返回的Tensor和原来的数据不再共享内存。

import numpy as np
a = np.ones((2,3))
c = torch.tensor(a)
a += 1
print('a:',a)
print('c:',c)
print(id(a)==id(c))

output：

a: [[2. 2. 2.]
 [2. 2. 2.]]
c: tensor([[1., 1., 1.],
        [1., 1., 1.]], dtype=torch.float64)
False