jax.numpy.ediff1d

jax.numpy.ediff1d(ary, to_end=None, to_begin=None)

数组中连续元素之间的差异。

LAX-backend 对 numpy.ediff1d() 的实现。

与 NumPy 的 ediff1d 实现不同，jax.numpy.ediff1d() 在将 to_end 或 to_begin 转换为 ary 的类型时，即使丢失精度也不会发出错误。

原始文档字符串如下。

参数:

• ary (array_like) – 如有必要，将在计算差异之前进行展平。

• to_end (array_like, optional) – 返回的差异末尾附加的数字。

• to_begin (array_like, optional) – 在返回的差异前添加的数字。

返回:

ediff1d – 差异。大致上，这相当于 ary.flat[1:] - ary.flat[:-1]。

返回类型:

ndarray