Sets mini-batch size. Called when layer is added to a network. :param batch_size: Mini-batch size.
(self, batch_size)
| 814 | ) # Synaptic input current decay (per timestep). |
| 815 | |
| 816 | def set_batch_size(self, batch_size) -> None: |
| 817 | # language=rst |
| 818 | """ |
| 819 | Sets mini-batch size. Called when layer is added to a network. |
| 820 | |
| 821 | :param batch_size: Mini-batch size. |
| 822 | """ |
| 823 | super().set_batch_size(batch_size=batch_size) |
| 824 | self.v = self.rest * torch.ones(batch_size, *self.shape, device=self.v.device) |
| 825 | self.i = torch.zeros_like(self.v, device=self.i.device) |
| 826 | self.refrac_count = torch.zeros_like(self.v, device=self.refrac_count.device) |
| 827 | |
| 828 | |
| 829 | class AdaptiveLIFNodes(Nodes): |
nothing calls this directly
no test coverage detected