fix tests

chaoming0625 · Sep 22, 2023 · 44adbc4 · 44adbc4
1 parent 0c46f00
commit 44adbc4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/brainpy/_src/dnn/normalization.py b/brainpy/_src/dnn/normalization.py
@@ -587,8 +587,8 @@ def update(self, x):
     x = (x - mean) * lax.rsqrt(var + lax.convert_element_type(self.epsilon, x.dtype))
     x = x.reshape(origin_shape)
     if self.affine:
-      x = x * lax.broadcast_to_rank(self.scale, origin_dim)
-      x = x + lax.broadcast_to_rank(self.bias, origin_dim)
+      x = x * lax.broadcast_to_rank(self.scale.value, origin_dim)
+      x = x + lax.broadcast_to_rank(self.bias.value, origin_dim)
     return x