maxframe.tensor.setdiff1d#

maxframe.tensor.setdiff1d(ar1, ar2, assume_unique=False)[源代码]#

找出两个张量的集合差集。

返回在 ar1 中但不在 ar2 中的唯一值。

参数:

  • ar1 (array_like) -- 输入张量。

  • ar2 (array_like) -- 输入比较张量。

  • assume_unique (bool) -- 如果为 True，则假定输入张量中的元素都是唯一的，这可以加速计算。默认值为 False。

返回:

setdiff1d -- 一个一维张量，包含在 ar1 中但不在 ar2 中的值。当 assume_unique=False 时，结果是排序的；否则仅在输入已排序时结果才是排序的。

返回类型:

Tensor

示例

>>> import maxframe.tensor as mt
>>> a = mt.array([1, 2, 3, 2, 4, 1])
>>> b = mt.array([3, 4, 5, 6])
>>> mt.setdiff1d(a, b).execute()
array([1, 2])