mars.tensor.equal#

mars.tensor.equal(x1, x2, out=None, where=None, **kwargs)[来源]#

逐元素返回 (x1 == x2)。

Parameters

  • x1 (array_like) – 相同形状的输入张量。

  • x2 (array_like) – 相同形状的输入张量。

  • out (张量, None, 或 元组 由 张量和 None 组成，可选) – 结果存储的位置。如果提供，必须具有与输入广播到的形状相同。如果未提供或 None，则返回新分配的数组。元组（仅作为关键字参数可能）必须具有与输出数量相等的长度。

  • where (array_like, 可选) – 值为 True 表示在该位置计算 ufunc，值为 False 表示保持输出中的该值不变。

  • **kwargs – 对于其他仅限关键字的参数，请参见 ufunc 文档.

Returns

out – 布尔值的输出张量，或者如果 x1 和 x2 是标量，则为单个布尔值。

Return type

张量或 bool

另请参阅

not_equal, greater_equal, less_equal, greater, less

示例

>>> import mars.tensor as mt

>>> mt.equal([0, 1, 3], mt.arange(3)).execute()
array([ True,  True, False])

比较的是值，而不是类型。因此，一个整数 (1) 和一个长度为一的张量可以评估为 True：

>>> mt.equal(1, mt.ones(1))
array([ True])