(self, x)
| 521 | self.eps = eps |
| 522 | |
| 523 | def initialize(self, x): |
| 524 | shape = (self.n_features,) |
| 525 | self.Gamma = Tensor(shape=shape, dtype=x.dtype, requires_grad=False, stores_grad=False) |
| 526 | self.Beta = Tensor(shape=shape, dtype=x.dtype, requires_grad=False, stores_grad=False) |
| 527 | self.Gamma.set_value(1.0) |
| 528 | self.Beta.set_value(0.0) |
| 529 | |
| 530 | def forward(self, x): |
| 531 | # x: input tensor with shape [batch_size, n_features] |