(self, inputs_shape)
| 473 | return params_shape, axes |
| 474 | |
| 475 | def build(self, inputs_shape): |
| 476 | params_shape, self.axes = self._get_param_shape(inputs_shape) |
| 477 | |
| 478 | self.beta, self.gamma = None, None |
| 479 | if self.beta_init: |
| 480 | self.beta = self._get_weights("beta", shape=params_shape, init=self.beta_init) |
| 481 | |
| 482 | if self.gamma_init: |
| 483 | self.gamma = self._get_weights("gamma", shape=params_shape, init=self.gamma_init) |
| 484 | |
| 485 | def forward(self, inputs): |
| 486 | mean, var = tf.nn.moments(inputs, self.axes, keepdims=True) |
no test coverage detected