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部分值方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持毛票票。
声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。