Tensor
张量转化为numpy
a = torch.FloatTensor(2,3)
print a.numpy(); # pytorch1.0版本以前
print a.detach().numpy(); # pytorch1.0
将numpy
转换为Tensor
张量
a = np.ones(5)
torch.from_numpy(a)
Tensor
张量转化为numpy
a = torch.FloatTensor(2,3)
print a.numpy(); # pytorch1.0版本以前
print a.detach().numpy(); # pytorch1.0
将numpy
转换为Tensor
张量
a = np.ones(5)
torch.from_numpy(a)