我需要将代表128x128 rgb图像的张量[1、3、128、128]转发到一个
nn.Conv2d(128,128,1)
但是我得到了错误
RuntimeError: Given groups=1, weight of size [128, 128, 1, 1], expected input[1, 3, 128, 128] to have 128 channels, but got 3 channels instead
PS：图像张量的形状为4d，因为上一步是nn.UpsamplingBilinear2d（size = None，scale_factor = 2），它需要一个4d张量作为输入