diff --git a/sonnet/src/bias.py b/sonnet/src/bias.py index 536c743f..43eb33f4 100644 --- a/sonnet/src/bias.py +++ b/sonnet/src/bias.py @@ -125,8 +125,7 @@ def __call__(self, inputs: tf.Tensor, multiplier: types.FloatLike = None): self._initialize(inputs) if multiplier is not None: return inputs + (self.b * multiplier) - else: - return inputs + self.b + return inputs + self.b def calculate_bias_shape(input_shape: types.ShapeLike,