2baaefbede4d08ab8f9080430d9a98f84a2d01e3,bindsnet/network/topology.py,MaxPool2dConnection,compute,#MaxPool2dConnection#Any#,400

Before Change

        B, C, H, W = indices.shape
        output = torch.zeros_like(indices)
        for b in range(B):
            for c in range(C):
                        output[b][c] = s[b][c].take(indices[b][c])

        return output.float()

    def update(self, **kwargs) -> None:
        // language=rst

After Change

            return_indices=True,
        )

        return s.flatten(2).gather(2, indices.flatten(2)).view_as(indices).float()

    def update(self, **kwargs) -> None:
        // language=rst
        

Instances