Sets mini-batch size. Called when layer is added to a network. :param batch_size: Mini-batch size.
(self, batch_size)
| 1305 | self.u = self.b * self.v # Neuron recovery. |
| 1306 | |
| 1307 | def set_batch_size(self, batch_size) -> None: |
| 1308 | # language=rst |
| 1309 | """ |
| 1310 | Sets mini-batch size. Called when layer is added to a network. |
| 1311 | |
| 1312 | :param batch_size: Mini-batch size. |
| 1313 | """ |
| 1314 | super().set_batch_size(batch_size=batch_size) |
| 1315 | self.v = self.rest * torch.ones(batch_size, *self.shape, device=self.v.device) |
| 1316 | self.u = self.b * self.v |
| 1317 | |
| 1318 | |
| 1319 | class CSRMNodes(Nodes): |
nothing calls this directly
no test coverage detected