mindspore.ops.floor_divide

查看源文件
mindspore.ops.floor_divide(input, other)[源代码]

逐元素计算 input 除以 other ,并向下取整。

如果 inputother 数据类型不同,遵循隐式类型转换规则。 输入必须是两个tensor或一个tensor和一个scalar。 当输入是两个tensor时,其shape须可以进行广播,并且数据类型不能同时为bool。

outi=floor(inputiotheri)

警告

这是一个实验性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.]