Pytorch之Tensor与NumPy数据类型转换

Tensor与NumPy数据类型基础转换

Tensor转NumPy

import torch
# input
x = torch.ones(5)
y = x.numpy()
# output
print(x)
tensor([1., 1., 1., 1., 1.])
print(y)
array([1., 1., 1., 1., 1.], dtype=float32)
print(x + 1)
tensor([2., 2., 2., 2., 2.])
print(y + 1)
[2. 2. 2. 2. 2.]

NumPy转Tensor(一)

import numpy as np
import torch
# input
x = np.ones(5)
y = torch.from_numpy(x)
# output
print(x)
array([1., 1., 1., 1., 1.])
print(y)
tensor([1., 1., 1., 1., 1.], dtype=torch.float64)
print(x + 1)
[2. 2. 2. 2. 2.]
print(y + 1)
tensor([2., 2., 2., 2., 2.], dtype=torch.float64)

NumPy转Tensor(二)

# input
x = np.ones(5)
z = torch.tensor(x)
# output
print(x)
array([1., 1., 1., 1., 1.])
print(z)
tensor([1., 1., 1., 1., 1.], dtype=torch.float64)
print(x + 1)
[2. 2. 2. 2. 2.]
print(z + 1)
tensor([2., 2., 2., 2., 2.], dtype=torch.float64)