python - 使用 Pythorch 翻转 2 张量蒙版

Question

我用 pytorch 创建了 2 个 2 个图像的蒙版，

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

口罩大小一样

我想翻转 2 个蒙版，将第一个蒙版放在第二张图像上，然后将第二个蒙版放在第一张图像上？

我可以这样做吗？

标签： pythonpytorchtensor