如何将以下以张量流编写的CNN转换为Pytorch中的模型？

Question

我有一个用Tensorflow编写的现有CNN网络模型。我正在将项目更改为pytorch。我是Pytorch的新手。我相信我在模型转换过程中所做的大部分工作都是正确的。但是我收到一个错误消息：无效的参数0：张量的大小必须匹配，除了3维。在1维中得到144和72

validate :selections_must_have_unique_option

  private

    def selections_must_have_unique_option
      unless self.choices.distinct(:option_id)
        errors.add(:options, 'can only have one choice per option')
      end
    end

这些在我的pytorch代码中的打印语句（print（x.size（）），print（pool4.size（）））打印为：torch.Size（[1，72，10，10]） torch.Size（[1，144，10，10]）。请帮助我进行此转移。谢谢。

Answer 1

您似乎正在尝试合并输出通道/功能图，这些输出通道/功能图在Pytorch中是尺寸1，而不是3。这可以在打印输出中看到。