mars.tensor.sign

mars.tensor.sign(x, out=None, where=None, **kwargs)

返回一个元素级的数字符号指示。

在sign函数中，如果x < 0, 将返回 -1，如果x==0, 将返回 0，如果x > 0，将返回 1。对于nan输入，将返回nan。

对于复杂输入，sign 函数返回 sign(x.real) + 0j 如果 x.real != 0 否则 sign(x.imag) + 0j。

对于复数 nan 输入，返回 complex(nan, 0)。

Parameters

• x (array_like) – 输入值。

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

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

• **kwargs –

Returns

y – x的符号。

Return type

张量

备注

对于复数，常见的正号有不止一种定义。这里使用的定义等同于 \(x/\sqrt{x*x}\)，这与一种常见的替代定义 \(x/|x|\) 不同。

示例

>>> import mars.tensor as mt

>>> mt.sign([-5., 4.5]).execute()
array([-1.,  1.])
>>> mt.sign(0).execute()
0
>>> mt.sign(5-2j).execute()
(1+0j)