mars.tensor.square#

mars.tensor.square(x, out=None, where=None, **kwargs)[来源]#

返回输入的逐元素平方。

Parameters

  • x (类数组) – 输入数据。

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

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

  • **kwargs –

Returns

out – 元素逐个 x*x，形状和数据类型与 x 相同。如果 x 是标量，则返回标量。

Return type

张量

另请参阅

sqrt, power

示例

>>> import mars.tensor as mt

>>> mt.square([-1j, 1]).execute()
array([-1.-0.j,  1.+0.j])