将语义分割的mask从torch.Size([ 224, 224])变为(torch.Size([ 2, 224, 224])代码

假设你的语义分割的mask张量为`mask`，可以使用以下代码将其转换为形状为`(2, 224, 224)`的张量：

import torch

# 假设mask为形状为(224, 224)的张量
mask = torch.randn(224, 224)

# 将mask转换为形状为(2, 224, 224)的张量
mask = torch.stack([1-mask, mask], dim=0)

这里使用了`torch.stack`函数，将`1-mask`和`mask`按照第0维进行堆叠，得到形状为`(2, 224, 224)`的张量。注意，这里我使用了随机生成的`mask`张量作为示例，实际应用中需要使用你的语义分割模型输出的张量。