1ee6d86439ca85e5c8d6bf5f8112668b8b63de77,bindsnet/conversion/conversion.py,SubtractiveResetIFNodes,__init__,#SubtractiveResetIFNodes#,89

Before Change

            self.refrac = refrac

        self.v = self.reset * torch.ones(self.shape)  // Neuron voltages.
        self.refrac_count = torch.zeros(self.shape)  // Refractory period counters.

    def forward(self, x: torch.Tensor) -> None:
        // language=rst

After Change

            "reset", torch.tensor(reset, dtype=torch.float)
        )  // Post-spike reset voltage.
        self.register_buffer(
            "thresh", torch.tensor(thresh, dtype=torch.float)
        )  // Spike threshold voltage.
        self.register_buffer(
            "refrac", torch.tensor(refrac)
        )  // Post-spike refractory period.

Instances