Pytorch 之修改Tensor部分值方式
时间：2021-02-17 14:03:26|栏目：Python代码|点击： 次
一：背景引入
对于一张图片，怎样修改局部像素值?
二：利用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。
栏 目：Python代码
本文地址：http://www.codeinn.net/misctech/64623.html
