(self)
| 12 | self.eps = eps |
| 13 | |
| 14 | def reset_parameters(self) -> None: |
| 15 | nn.init.constant_(self.weight, 1) |
| 16 | |
| 17 | def _norm(self, x: Tensor) -> Tensor: |
| 18 | return x * torch.rsqrt(x.pow(2).mean(-1, keepdim=True) + self.eps) |
nothing calls this directly
no outgoing calls
no test coverage detected