如何解决pytorch操作:火炬张量中值的并集和求逆
在cuda / GPU上我有一组与图像中不同对象相对应的蒙版(以下形状和大小)
masks.shape: torch.Size([10,240,320,1])
masks[0].shape: torch.Size([240,1])
masks[0][:,:,0].shape: torch.Size([240,320])
1:我可以使用torch.tensor操作产生这些面罩的并集吗?这样我就可以将所有这些图像一次应用到图像上?
2:如何反转火炬张量中的值?我的意思是1,变成0,反之亦然。我试图〜mytensor,但它说运算符仅适用于整数或布尔值。我的张量中有浮点值,例如[1.]等。
我打算在GPU上执行所有这些操作,而又不将数据移回cpu。
谢谢。
解决方法
2:如何反转火炬张量中的值?
t = torch.tensor([1.,0.,1.,1.])
如果您不想更改类型,则可以从1中减去值。
1 - t
tensor([0.,0.])
或者更好的是,您可以将其转换为布尔类型,然后使用~
~t.type(torch.bool)
tensor([False,True,False,False])
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。