49dc427e2c196a334dc7b173ff317e05f04813e7,tests/concat_many.py,TestConcatMany,forward,#TestConcatMany#,23

Before Change

            [nn.Conv2d(inp, out, kernel_size=kernel_size, bias=bias) for i in range(k)]

    def forward(self, x):
        conv2ds_results = [c(x) for c in self.conv2ds]
        x = torch.cat(conv2ds_results, dim=1)
        return x


if __name__ == "__main__":

After Change

        self.conv2_3 = nn.Conv2d(inp, out, kernel_size=kernel_size, bias=bias)

    def forward(self, x):
        x = torch.cat([
            self.conv2_1(x),
            self.conv2_2(x),
            self.conv2_3(x)
        ], dim=1)
        return x


if __name__ == "__main__":

Instances