2179685be9f876bc33362d0c680dbd68c3c4de0a,bindsnet/learning/learning.py,PostPre,_connection_update,#PostPre#,172

Before Change

        // Pre-synaptic update.
        if self.nu[0]:
            update = self.reduction(torch.bmm(source_s, target_x), dim=0)
            self.connection.w -= self.nu[0] * update

        // Post-synaptic update.
        if self.nu[1]:

After Change

        if self.nu[0]:
            source_s = self.source.s.view(batch_size, -1).unsqueeze(2).float()
            target_x = self.target.x.view(batch_size, -1).unsqueeze(1) * self.nu[0]
            self.connection.w -= self.reduction(torch.bmm(source_s, target_x), dim=0)
            del source_s, target_x

        // Post-synaptic update.
        if self.nu[1]:

Instances