[python][pytorch]torch.Tensor数值类型转换

a = torch.tensor([1,2,3])
b = a.float()
c = a.double()
d = a.long()
print(b.dtype)
print(c.dtype)
print(d.dtype)



 

猜你喜欢

转载自blog.csdn.net/FL1623863129/article/details/132436458