mars.tensor.sqrt

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

返回张量的正平方根，逐元素计算。

Parameters

• x (array_like) – 需要计算平方根的值。

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

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

• **kwargs –

Returns

y – 一个形状与 x 相同的张量，包含 x 中每个元素的正平方根。如果 x 中的任何元素是复数，将返回一个复数张量（并计算负实数的平方根）。如果 x 中的所有元素都是实数，则 y 也是实数，负元素返回 nan。如果提供了 out，则 y 是对它的引用。

Return type

张量

备注

sqrt 一般遵循常见约定，其支路切割为实数“区间” [-inf, 0)，并且在该区间上从上方连续。支路切割是复平面上的一条曲线，穿越该曲线后，给定的复函数不再连续。

示例

>>> import mars.tensor as mt

>>> mt.sqrt([1,4,9]).execute()
array([ 1.,  2.,  3.])

>>> mt.sqrt([4, -1, -3+4J]).execute()
array([ 2.+0.j,  0.+1.j,  1.+2.j])

>>> mt.sqrt([4, -1, mt.inf]).execute()
array([  2.,  NaN,  Inf])