pytorch异常——RuntimeError:Given groups=1, weight of size..., expected of...
文章目录省流异常报错异常截图异常代码原因解释修正代码执行结果省流nn.Conv2d需要的输入张量格式为(batch_size,channels,height,width),但您的示例输入张量x是(batch_size,height,width,channels)。因此,需要对输入张量进行转置。注意,TensorFlow使用"NHWC"(批次、高度、宽度、通道)格式,而PyTorch使用"NCHW"