单通道tensor转化为三通道tensor的代码

可以使用torch.stack()函数将单通道tensor复制三份，然后在第三个维度上进行拼接，代码如下：

import torch

# 假设单通道tensor为x，大小为[batch_size, height, width]
x = torch.randn(batch_size, height, width)

# 将单通道tensor复制三份
x = torch.stack([x, x, x], dim=1)

# 在第三个维度上进行拼接
x = torch.cat([x, x, x], dim=2)

# 得到三通道tensor，大小为[batch_size, 3, height, width]
print(x.size())