Skip to content

Commit

Permalink
fix (#182)
Browse files Browse the repository at this point in the history
  • Loading branch information
zbt78 committed Feb 5, 2024
1 parent f6859c4 commit f62b46f
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ def batch_p_dist(x, y, p=2):
"""
x = x.unsqueeze(1)
diff = x - y
return paddle.norm(diff, p=p, axis=list(range(2, diff.dim())))
return paddle.linalg.vector_norm(diff, p=p, axis=list(range(2, diff.dim())))


@register
Expand Down

0 comments on commit f62b46f

Please sign in to comment.