mars.tensor.rint#

mars.tensor.rint(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

out – 输出数组与 x 形状和类型相同。

Return type

张量或标量

另请参阅

ceil, floor, trunc

示例

>>> import mars.tensor as mt

>>> a = mt.array([-1.7, -1.5, -0.2, 0.2, 1.5, 1.7, 2.0])
>>> mt.rint(a).execute()
array([-2., -2., -0.,  0.,  2.,  2.,  2.])