fix aster loss for axis (#8675)

2026-09-21 13:23:18 +08:00 · 2022-12-20 13:58:57 +08:00
parent 582e868cf8
commit ca8de0c69e
1 changed files with 1 additions and 1 deletions
@@ -28,7 +28,7 @@ class CosineEmbeddingLoss(nn.Layer):

    def forward(self, x1, x2, target):
        similarity = paddle.sum(
-            x1 * x2, dim=-1) / (paddle.norm(
+            x1 * x2, axis=-1) / (paddle.norm(
                x1, axis=-1) * paddle.norm(
                    x2, axis=-1) + self.epsilon)
        one_list = paddle.full_like(target, fill_value=1)