我有一个224×224的二进制图像，存储在一个张量中（1, 224, 224），其中0像素表示背景，1像素表示前景。我希望将其重塑为一个张量（2, 224, 224），使得第一层gt[0]在原图像中0像素的地方显示1像素，反之亦然。这样，一层应该在背景处显示1，而另一层则在前景处显示1（基本上，我需要在这个张量中有两个互补的二进制图像）。

这是我的代码：

# gt是一个张量（1, 224, 224）
gt = gt.expand((2, 224, 224))
backgr = gt[0]
foregr = gt[1]
backgr[backgr == 0] = 2 # 将所有0换成1，反之亦然
backgr[backgr == 1] = 0
backgr[backgr == 2] = 1
gt[0] = backgr
print(gt[0])
print(gt[1])

问题是使用这段代码两个层都被修改了，我无法弄清楚如何保持其中一个层不变，只修改gt[0]。

回答：

找到了解决方案！

gt = gt.repeat(2, 1, 1)