pytorch 操作：火炬张量中值的并集和反转

Question

我在 cuda/GPU 上有一组对应于图像中不同对象的掩码（下面的形状和大小）

masks.shape:            torch.Size([10, 240, 320, 1])
masks[0].shape:         torch.Size([240, 320, 1])
masks[0][:,:,0].shape:  torch.Size([240, 320])

1：我可以使用 torch.tensor 操作生成这些面具的并集吗？ 以便我可以在图像上应用所有这些一次？

2：如何反转火炬张量中的值？ 我的意思是 1，变成 0，反之亦然。 我试过 ~mytensor 但它说该运算符仅适用于整数或布尔值。 我的张量中有浮点值，即 [1.] 等。

我打算在 GPU 上执行所有这些操作，而不将数据移回 CPU。

谢谢你。

Answer 1

2：如何反转火炬张量中的值？

t = torch.tensor([1., 0., 0., 1., 0., 1., 1.])

如果您不想更改类型，可以从 1 中减去值

1 - t
tensor([0., 1., 1., 0., 1., 0., 0.])

或者更好的是，您可以将其转换为布尔类型，然后使用~

~t.type(torch.bool)
tensor([False,  True,  True, False,  True, False, False])