pytorch tensor 类型转换 (需求:int64转成tensor.inte64)
决定开始记录自己做项目过程中遇到的bug啦。
我利用torch.Tensor()转换int64类型的数据,转完之后tensor中元素为浮点数。一番x.dytpe查看元素具体类型后才发现torch.Tensor()默认转成tensor.float32,如果要tensor.int64,需要LongTensor。
所以没事多看官方文档很重要。
决定开始记录自己做项目过程中遇到的bug啦。
我利用torch.Tensor()转换int64类型的数据,转完之后tensor中元素为浮点数。一番x.dytpe查看元素具体类型后才发现torch.Tensor()默认转成tensor.float32,如果要tensor.int64,需要LongTensor。
所以没事多看官方文档很重要。
相关文章: