Pytorch 之修改Tensor部分值方式
阿新 • • 發佈：2020-01-08
一：背景引入
對於一張圖片，怎樣修改區域性畫素值?
二：利用Tensor方法
比如輸入全零tensor，可認為為黑色圖片
>>> n=torch.FloatTensor(3,3,4).fill_(0)
>>> n
tensor([[[0.,0.,0.],[0.,0.]],[[0.,0.]]])
現在修改部分值：
>>> n[:,0:2,1:3]=1
結果為
>>> n
tensor([[[0.,1.,0.]]])
可以看到利用切片的思想可以修改tensor中的部分值。這樣修改後，黑色圖片上就會有白色的小洞洞，因為在tensor中，1代表是白色，RGB中的255。
以上這篇Pytorch 之修改Tensor部分值方式就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支援我們。