【torch中的tensor转化为numpy张量】

torch中的tensor转化为numpy张量

# torch中的tensor转化为numpy张量
X = torch.arange(12, dtype=torch.float32).reshape((3,4))
Y = torch.tensor([[2.0, 1, 4, 3], [1, 2, 3, 4], [4, 3, 2, 1]])

A = X.numpy()
B = torch.tensor(A)
type(A), type(B)

将大下为1的张量转化为python标量

# 将大下为1的张量转化为python标量
a = torch.tensor([5.7])
a,a.item(), float(a), int(a)