pytorch tensor 类型转换 (需求:int64转成tensor.inte64)

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