mindspore.ops.floor_divide
- mindspore.ops.floor_divide(input, other)[源代码]
逐元素计算 input 除以 other ,并向下取整。
如果 input 和 other 数据类型不同,遵循隐式类型转换规则。 输入必须是两个tensor或一个tensor和一个scalar。 当输入是两个tensor时,其shape须可以进行广播,并且数据类型不能同时为bool。
警告
这是一个实验性API,后续可能修改或删除。
- 参数:
input (Union[Tensor, Number, bool]) - 第一个输入tensor。
other (Union[Tensor, Number, bool]) - 第二个输入tensor。
- 返回:
Tensor
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> # case 1: Two tensors with boolean and integer data type. >>> input = mindspore.tensor([True, True, False]) >>> other = mindspore.tensor([1, 2, 4]) >>> output = mindspore.ops.floor_divide(input, other) >>> print(output) [1 0 0] >>> >>> # case 2: One tensor and one scalar. >>> input = mindspore.tensor([1, 2, 4]) >>> other = mindspore.tensor(1.5) >>> output = mindspore.ops.floor_divide(input, other) >>> print(output) [0. 1. 2.] >>> >>> # case 3: When inputs have different data types, type promotion rules are followed. >>> input = mindspore.tensor([1, 2, 4], mindspore.int32) >>> other = mindspore.tensor([1.1, 2.5, -1.5], mindspore.float32) >>> output = mindspore.ops.floor_divide(input, other) >>> print(output) [ 0. 0. -3.]