Python 用Pythorch翻转2张量蒙版

我用pytorch制作了两个面具，包括两幅图像

self.img_m = self.mask * self.img_truth
x = self.mask * self.img_m
y = (1 - self.mask) * self.img_g[-1].detach()

这些面具大小相同

我想翻转两个遮罩，将第一个遮罩放在第二个图像上，并将第二个遮罩放在第一个图像上

---

我可以这样做吗？

图像的大小也一样吗？