Pytorch数据类型转换

载入模块生成数据

import torch

import numpy as np

a_numpy = np.array([1,2,3])

a_tensor = torch.from_numpy(a_numpy)

print(a_tensor)

a_numpy = a_tensor.numpy()

print(a_numpy)

c = torch.tensor(2)

print(c)

c = c.item()

print(c)

a_variable = Variable(torch.from_numpy(a_numpy))

print(a_variable)

a_numpy = a_variable.data.numpy()

print(a_numpy)