maxframe.tensor.ediff1d#

maxframe.tensor.ediff1d(a, to_end=None, to_begin=None)[源代码]#

张量中连续元素之间的差值。

参数:

  • a (array_like) -- 如有必要，将在计算差值之前将其展平。

  • to_end (array_like, optional) -- 附加在返回差值末尾的数字。

  • to_begin (array_like, optional) -- 前置在返回差值开头的数字。

返回:

ediff1d -- 差值。大致上，这是 a.flat[1:] - a.flat[:-1]。

返回类型:

Tensor

参见

diff, gradient

示例

>>> import maxframe.tensor as mt

>>> x = mt.array([1, 2, 4, 7, 0])
>>> mt.ediff1d(x).execute()
array([ 1,  2,  3, -7])

>>> mt.ediff1d(x, to_begin=-99, to_end=mt.array([88, 99])).execute()
array([-99,   1,   2,   3,  -7,  88,  99])

返回的张量始终是一维的。

>>> y = [[1, 2, 4], [1, 6, 24]]
>>> mt.ediff1d(y).execute()
array([ 1,  2, -3,  5, 18])