Pytorch tensor类型转换中的深坑

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

另外记录一下tensor的类型转化方法:dtype=troch.int32
97dc6288078e49e2994b50f52f113683