c3498e8064176494925856449a4aab2c45701c09,bindsnet/network/nodes.py,IzhikevichNodes,forward,#IzhikevichNodes#Any#,1205

Before Change

        // Add inter-columnar input.
        if self.s.any():
            x += self.S[:, self.s.view(-1)].sum(dim=1)

        // Apply v and u updates.
        self.v += self.dt * 0.5 * (0.04 * self.v ** 2 + 5 * self.v + 140 - self.u + x)

After Change

        // Add inter-columnar input.
        if self.s.any():
            x += torch.cat([self.S[:, self.s[i]].sum(dim=1)[None] for i in range(self.s.shape[0])], dim=0)

        // Apply v and u updates.
        self.v += self.dt * 0.5 * (0.04 * self.v ** 2 + 5 * self.v + 140 - self.u + x)

Instances