pythonJax小记（四）：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