小编Imr*_*ran的帖子

我想创建和训练自动编码器来提取特征并将该特征用于聚类算法。现在我在计算损失时遇到错误。

运行时错误：张量 a (224) 的大小必须与非单一维度 3 处的张量 b (244) 的大小匹配

和一个警告

用户警告：使用与输入大小 (torch.Size([1, 3, 224, 224])) 不同的目标大小 (torch.Size([1, 3, 224, 244]))。这可能会因广播而导致错误的结果。请确保它们的尺寸相同。

• 返回F.mse_loss（输入，目标，减少= self.reduction）*

我正在使用 Pytorch。

谁能告诉我这有什么问题吗？在警告和错误中，输入和输出的大小是相同的，但它是不同的。输入和输出图像的汇总大小如下

----------------------------------------------------------------
        Layer (type)               Output Shape         Param #
================================================================
            Conv2d-1         [-1, 16, 112, 112]             448
              ReLU-2         [-1, 16, 112, 112]               0
            Conv2d-3           [-1, 32, 56, 56]           4,640
              ReLU-4           [-1, 32, 56, 56]               0
            Conv2d-5           [-1, 64, 18, 18]         100,416
              ReLU-6           [-1, 64, 18, 18]               0
            Conv2d-7            [-1, 128, 3, 3]         401,536
              ReLU-8            [-1, 128, 3, …