今天在写代码时遇到了一个奇怪的问题,当我定义一个全部为整型的tensor变量,我发现其中的元素会自动转换成浮点型:
找了很久原因才发现是方法名有着极微的差别:
torch.Tensor()是用来定义浮点型tensor类型的,也就是说不管你里面元素填的什么类型,最终都会变成浮点型
而torch.tensor() 注意是小写t,才是用来定义tensor类型变量的,你里面写什么类型的元素他就是什么类型的
另外记录一下tensor的类型转化方法:dtype=troch.int32
今天在写代码时遇到了一个奇怪的问题,当我定义一个全部为整型的tensor变量,我发现其中的元素会自动转换成浮点型:
找了很久原因才发现是方法名有着极微的差别:
torch.Tensor()是用来定义浮点型tensor类型的,也就是说不管你里面元素填的什么类型,最终都会变成浮点型
而torch.tensor() 注意是小写t,才是用来定义tensor类型变量的,你里面写什么类型的元素他就是什么类型的
另外记录一下tensor的类型转化方法:dtype=troch.int32