溫馨提示×

您好,登錄后才能下訂單哦!

密碼登錄×
登錄注冊(cè)×
其他方式登錄
點(diǎn)擊 登錄注冊(cè) 即表示同意《億速云用戶服務(wù)條款》

如何在Pytorch中修改Tensor的部分值

發(fā)布時(shí)間:2021-03-16 15:45:01 來源:億速云 閱讀:1723 作者:Leah 欄目:開發(fā)技術(shù)

今天就跟大家聊聊有關(guān)如何在Pytorch中修改Tensor的部分值,可能很多人都不太了解,為了讓大家更加了解,小編給大家總結(jié)了以下內(nèi)容,希望大家根據(jù)這篇文章可以有所收獲。

利用Tensor方法

比如輸入全零tensor,可認(rèn)為為黑色圖片

>>> n=torch.FloatTensor(3,3,4).fill_(0)
>>> n
    tensor([[[0., 0., 0., 0.], 
         [0., 0., 0., 0.],                          
         [0., 0., 0., 0.]], 
         
         [[0., 0., 0., 0.], 
         [0., 0., 0., 0.],
         [0., 0., 0., 0.]], 
 
         [[0., 0., 0., 0.], 
         [0., 0., 0., 0.],
         [0., 0., 0., 0.]]])

現(xiàn)在修改部分值:

>>> n[:,0:2,1:3]=1

結(jié)果為

>>> n
    tensor([[[0., 1., 1., 0.], 
         [0., 1., 1., 0.],                          
         [0., 0., 0., 0.]], 
         
         [[0., 1., 1., 0.], 
         [0., 1., 1., 0.],
         [0., 0., 0., 0.]], 
 
         [[0., 1., 1., 0.], 
         [0., 1., 1., 0.],
         [0., 0., 0., 0.]]])

可以看到利用切片的思想可以修改tensor中的部分值。這樣修改后,黑色圖片上就會(huì)有白色的小洞洞,因?yàn)樵趖ensor中,1代表是白色,RGB中的255。

看完上述內(nèi)容,你們對(duì)如何在Pytorch中修改Tensor的部分值有進(jìn)一步的了解嗎?如果還想了解更多知識(shí)或者相關(guān)內(nèi)容,請(qǐng)關(guān)注億速云行業(yè)資訊頻道,感謝大家的支持。

向AI問一下細(xì)節(jié)

免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng),如果涉及侵權(quán)請(qǐng)聯(lián)系站長(zhǎng)郵箱:is@yisu.com進(jìn)行舉報(bào),并提供相關(guān)證據(jù),一經(jīng)查實(shí),將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI