(self, input, idx)
| 335 | init.zeros_(self.bias) |
| 336 | |
| 337 | def forward(self, input, idx): |
| 338 | if self.elementwise_affine: |
| 339 | return F.layer_norm(input, tuple(input.shape[1:]), self.weight[:,idx,:], self.bias[:,idx,:], self.eps) |
| 340 | else: |
| 341 | return F.layer_norm(input, tuple(input.shape[1:]), self.weight, self.bias, self.eps) |
| 342 | |
| 343 | def extra_repr(self): |
| 344 | return '{normalized_shape}, eps={eps}, ' \ |
nothing calls this directly
no outgoing calls
no test coverage detected