pytorch tensor 類型轉換 (需求:int64轉成tensor.inte64)

pytorch tensor 類型轉換 (需求:int64轉成tensor.inte64) 決定開始記錄自己做項目過程中遇到的bug啦。 我利用torch.Tensor()轉換int64類型的數據,轉完之後tensor中元素爲浮點數。一番x.dytpe查看元素具體類型後才發現torch.Tensor()默認轉成tensor.float32,如果要tensor.int64,需要LongTensor。
相關文章
相關標籤/搜索