pythonJax小记(四):python: 使用Jax计算两个数组之间的欧几里得距离(持续更新,评论区可以补充)

python: 使用Jax计算两个数组之间的欧几里得距离


前言

自用,刚开始接触可能顺序会比较乱。


直接上代码

import jax.numpy as jnp

array1 = jnp.array([4, 5, 7])
array2 = jnp.array([4, 5, 6])
 
# 计算欧几里得距离
distance = jnp.linalg.norm(array1 - array2)
 
print("Euclidean Distance:", distance)

输出:

Euclidean Distance: 1.0

猜你喜欢

转载自blog.csdn.net/xzs1210652636/article/details/136274752