mindspore.ops.floor

查看源文件
mindspore.ops.floor(input)[源代码]

逐元素向下取整函数。

\[out_i = \lfloor input_i \rfloor\]
参数:
  • input (Tensor) - 输入Tensor。支持的数据类型:

    • Ascend: float16、float32、float64、bfloat16、int8、int16、int32、int64、uint8、uint16、uint32、uint64。

    • GPU/CPU: float16、float32、float64。

返回:

Tensor,shape与 input 相同。

异常:
  • TypeError - input 不是Tensor。

  • TypeError - input 的数据类型不支持。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> input = Tensor(np.array([1.1, 2.5, -1.5]), mindspore.float32)
>>> output = ops.floor(input)
>>> print(output)
[ 1.  2. -2.]