mindspore.ops.TruncateDiv

查看源文件
class mindspore.ops.TruncateDiv[源代码]

将第一个输入Tensor与第二个输入Tensor逐元素相除，结果将向0取整。相当于C语言风格的整数除法。

输入 x 和 y 应能遵循隐式类型转换规则使数据类型一致。 当输入为两个Tensor时，数据类型不能同时为bool类型。 当输入是一个Tensor和一个标量时，标量只能是一个常数。

说明

支持shape广播。

输入：

  • x (Union[Tensor, Number, bool]) - Number或bool类型的Tensor。

  • y (Union[Tensor, Number, bool]) - Number或bool类型的Tensor。x 和 y 不能同时为bool类型。

输出：

Tensor，shape为输入广播后的shape，数据类型为两个输入中精度较高的输入的类型。

异常：

  • TypeError - x 和 y 数据类型不是以下之一：Tensor、Number或bool。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> x = Tensor(np.array([2, 4, -1]), mindspore.int32)
>>> y = Tensor(np.array([3, 3, 3]), mindspore.int32)
>>> truncate_div = ops.TruncateDiv()
>>> output = truncate_div(x, y)
>>> print(output)
[0 1 0]