关于灰度图像DCGAN (Pytorch)中通道大小的问题

import torchvision.transforms as transforms

# 将图像转换为灰度图像，并将通道数设置为1
transform = transforms.Compose([
    transforms.Grayscale(num_output_channels=1),
    transforms.ToTensor()
])

# 加载灰度图像
gray_image = transform(image)

import torch.nn as nn

class Generator(nn.Module):
    def __init__(self, input_channels, output_channels):
        super(Generator, self).__init__()
        self.model = nn.Sequential(
            # 输入通道数为1，输出通道数为64
            nn.ConvTranspose2d(input_channels, 64, kernel_size=4, stride=2, padding=1),
            nn.ReLU(),
            # ...
            # 输出通道数为3，生成RGB彩色图像
            nn.ConvTranspose2d(64, output_channels, kernel_size=4, stride=2, padding=1),
            nn.Tanh()
        )

    def forward(self, x):
        return self.model(x)