1.查看數據類型
常用類型有 : torch.IntTensor、 torch.FloatTensor
torch.Tensor是默認的tensor類型(torch.FloatTensor)的簡稱
tensor.dtype
2.類型轉換
方法一:簡單后綴轉換
tensor.int()
tensor.float()
tensor.double()
方法二:使用torch.type()函數
tensor.type(torch.FloatTensor)
方法三:使用type_as(tensor)將tensor轉換為指定tensor的類型
3.tensor創建--指定維度和數據類型
torch.IntTensor(3,4).zero_()
torch.Tensor(3,4).zero_()