pytorch tensor类型转换中的深坑
今天在写代码时遇到了一个奇怪的问题,当我定义一个全部为整型的tensor变量,我发现其中的元素会自动转换成浮点型:
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/743d80ba16dd951f743aaceccbaeab8f.png)
找了很久原因才发现是方法名有着极微的差别:
torch.Tensor()是用来定义浮点型tensor类型的,也就是说不管你里面元素填的什么类型,最终都会变成浮点型
而torch.tensor() 注意是小写t,才是用来定义tensor类型变量的,你里面写什么类型的元素他就是什么类型的
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/ba54c1ce46c72a1c5dbb33e162a8c509.png)
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/0da3655c378a713e8de0438b703f242a.png)
另外记录一下tensor的类型转化方法:dtype=troch.int32