(self, inputs_shape)
| 53 | return s.format(classname=self.__class__.__name__, **self.__dict__) |
| 54 | |
| 55 | def build(self, inputs_shape): |
| 56 | self.scale = self._get_weights("scale", shape=[1], init=constant(value=self.init_scale)) |
| 57 | |
| 58 | # @tf.function |
| 59 | def forward(self, inputs): |