jax.numpy.diff

jax.numpy.diff(a, n=1, axis=-1, prepend=None, append=None)

计算沿给定轴的第 n 次离散差分。

LAX 后端实现 numpy.diff()。

原始文档字符串如下。

第一个差异由沿给定轴的 out[i] = a[i+1] - a[i] 给出，更高的差异通过递归使用 diff 计算。

参数:

• a (array_like) – 输入数组

• n (int, optional) – 值被差分的次数。如果为零，则输入保持原样返回。

• axis (int, optional) – 差异计算所沿的轴，默认是最后一个轴。

• prepend (array_like, optional) – 在沿轴执行差分之前，要添加到 a 的值。标量值会沿轴方向扩展为长度为1的数组，并在所有其他轴上与输入数组的形状匹配。否则，维度与形状必须与 a 匹配，除了沿轴的维度。

• append (array_like, optional) – 在沿轴执行差分之前，要添加到 a 的值。标量值会沿轴方向扩展为长度为1的数组，并在所有其他轴上与输入数组的形状匹配。否则，维度与形状必须与 a 匹配，除了沿轴的维度。

返回:

diff – 第 n 次差分。输出的形状与 a 相同，除了沿 axis 的维度会减少 n。输出的类型与 a 中任意两个元素的差值类型相同。在大多数情况下，这与 a 的类型相同。一个显著的例外是 datetime64，它会导致一个 timedelta64 输出数组。

返回类型:

ndarray