(self)
| 267 | self.num_batches_tracked.zero_() |
| 268 | |
| 269 | def reset_parameters(self): |
| 270 | self.reset_running_stats() |
| 271 | if self.affine: |
| 272 | self.Br.data.zero_() |
| 273 | self.Bi.data.zero_() |
| 274 | self.Wrr.data.fill_(1) |
| 275 | self.Wri.data.uniform_(-.9, +.9) # W will be positive-definite |
| 276 | self.Wii.data.fill_(1) |
| 277 | |
| 278 | def _check_input_dim(self, xr, xi): |
| 279 | assert(xr.shape == xi.shape) |
no test coverage detected