如何在Pytorch中修改Tensor的部分值-創(chuàng)新互聯
今天就跟大家聊聊有關如何在Pytorch中修改Tensor的部分值，可能很多人都不太了解，為了讓大家更加了解，小編給大家總結了以下內容，希望大家根據這篇文章可以有所收獲。
利用Tensor方法
比如輸入全零tensor，可認為為黑色圖片
>>> 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.]]])
現在修改部分值：
>>> n[:,0:2,1:3]=1
結果為
>>> 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中的部分值。這樣修改后，黑色圖片上就會有白色的小洞洞，因為在tensor中，1代表是白色，RGB中的255。
看完上述內容，你們對如何在Pytorch中修改Tensor的部分值有進一步的了解嗎？如果還想了解更多知識或者相關內容，請關注創(chuàng)新互聯行業(yè)資訊頻道，感謝大家的支持。
網頁名稱：如何在Pytorch中修改Tensor的部分值-創(chuàng)新互聯
路徑分享：http://m.jiaotiyi.com/article/csodci.html