du = torch.ones([2,2])
a = np.array([[1,2],[3,4]],dtype=np.float32)
b = torch.from_numpy(a)#數(shù)據(jù)類(lèi)型是不變的 上面dtype是什么類(lèi)型 下面torch對(duì)應(yīng)就是什么tensor
print('yes')
#tensor 默認(rèn)是float32的 比如torch.ones zeros都是生成float32數(shù)據(jù)a=torch.FloatTensor( (3,2) )
print(a.dtype)
qq = np.array([1,2],dtype=np.int32)
qq = qq.astype(np.float32) #這個(gè)是numpy的類(lèi)型轉(zhuǎn)換 下面是tensor的類(lèi)型轉(zhuǎn)換 只astype或者int()并不能改變數(shù)據(jù)類(lèi)型
#必須有賦值操作才可以a = a.int()##tensor 數(shù)據(jù)類(lèi)型轉(zhuǎn)換
print(a.dtype)#對(duì)于python的數(shù)據(jù) 直接進(jìn)行類(lèi)型轉(zhuǎn)化即可
#例如
m = 5
n = float(m)

torch 數(shù)據(jù)類(lèi)型轉(zhuǎn)換：
tensor = torch.Tensor(3, 5)# torch.long() 將tensor投射為long類(lèi)型
newtensor = tensor.long()# torch.half()將tensor投射為半精度浮點(diǎn)類(lèi)型
newtensor = tensor.half()# torch.int()將該tensor投射為int類(lèi)型
newtensor = tensor.int()# torch.double()將該tensor投射為double類(lèi)型
newtensor = tensor.double()# torch.float()將該tensor投射為float類(lèi)型
newtensor = tensor.float()# torch.char()將該tensor投射為char類(lèi)型
newtensor = tensor.char()# torch.byte()將該tensor投射為byte類(lèi)型
newtensor = tensor.byte()# torch.short()將該tensor投射為short類(lèi)型
newtensor = tensor.short()

總結(jié)

以上是生活随笔為你收集整理的数据类型转换pytorch的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。