mars.tensor.setdiff1d#

mars.tensor.setdiff1d(ar1, ar2, assume_unique=False)[来源]#

找到两个张量的集合差异。

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

Parameters

  • ar1 (array_like) – 输入张量。

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

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

Returns

setdiff1d – 在 ar1 中但不在 ar2 中的值的 1D 张量。结果在 assume_unique=False 时会被排序，但在输入未排序时，结果也仅在输入已排序时才会被排序。

Return type

张量

示例

>>> import mars.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])