981e912036e667a0d0e100a12001e3e8543620ea,pytorch/pytorchcv/models/sknet.py,SKConvBlock,forward,#SKConvBlock#Any#,70

Before Change

        w = self.softmax(w)

        w = list(w.chunk(self.num_branches, dim=1))
        w = list(map(lambda y: y.reshape(batch, self.out_channels, 1, 1), w))
        v = list(map(lambda y1, y2: y1 * y2, outs, w))
        v = reduce(lambda y1, y2: y1 + y2, v)
        return v

After Change

    def forward(self, x):
        y = self.branches(x)

        u = y.sum(dim=1)
        s = self.pool(u)
        z = self.fc1(s)
        w = self.fc2(z)

Instances