关于tensor中数值类型

X = torch.tensor([[1,2],[3,4]])
#查看X的类型
print(type(X))
#查看数据值的类型 
print(X.dtype)
#X的数据类型为torch.int64,因为输入的数据全为整数

Y1 = torch.ones(2,2)
Y2 = torch.zeros(2,2)
Y3 = torch.randn(2,2)
#Y1、Y2、Y3的数据类型为torch.float32,torch自带的定义张量方式指均为float型

你可能感兴趣的:(Pytorch)