mindspore.ops.floor_divide
- mindspore.ops.floor_divide(input, other)[源代码]
按元素将第一个输入Tensor除以第二个输入Tensor,并向下取整。
input 和 other 的输入遵循隐式类型转换规则,使数据类型一致。输入必须是两个Tensor或一个Tensor和一个Scalar。当输入是两个Tensor时,它们的数据类型不能同时为bool,其shape可以广播。当输入是一个Tensor和一个Scalar时,Scalar只能是一个常量。
\[out_{i} = \text{floor}( \frac{input_i}{other_i})\]警告
这是一个实验性API,后续可能修改或删除。
其中 \(floor\) 表示Floor算子,有关更多详细信息,请参阅
mindspore.ops.Floor
算子。- 参数:
input (Union[Tensor, Number, bool]) - 第一个输入,为数值型,或bool,或数据类型为数值型或bool的Tensor。
other (Union[Tensor, Number, bool]) - 第二个输入,当第一个输入是Tensor时,第二个输入必须是一个数值型或bool,或是数据类型为数值型或bool的Tensor。
- 返回:
Tensor,输出的shape与广播后的shape相同,数据类型取两个输入中精度较高或数字较高的。
- 异常:
TypeError - 如果 input 和 other 都不是Tensor。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> from mindspore import Tensor, ops >>> import numpy as np >>> x = Tensor(np.array([2, 4, -1]), mindspore.int32) >>> y = Tensor(np.array([3, 3, 3]), mindspore.int32) >>> output = ops.floor_divide(x, y) >>> print(output) [ 0 1 -1] >>> x = Tensor(2.0, mindspore.float32) >>> y = Tensor(2.0, mindspore.float32) >>> output = ops.floor_divide(x, y) >>> print(output) 1.0