mindspore.ops.maximum

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

逐元素计算两个输入tensor中最大值。

outputi=max(inputi,otheri)

说明

  • 输入 inputother 遵循隐式类型转换规则,使数据类型保持一致。

  • 当输入是两个Tensor时,它们的数据类型不能同时是bool,并保证其shape可以广播。

  • 当输入是一个Tensor和一个Scalar时,Scalar只能是一个常数。

  • 支持广播。

  • 如果一个元素和NaN比较,则返回NaN。

警告

如果所有输入都为标量int类型,在Graph模式下,输出为int32类型的Tensor,在PyNative模式下,输出为int64类型的Tensor。

参数:
  • input (Union[Tensor, Number, bool]) - 第一个输入。

  • other (Union[Tensor, Number, bool]) - 第二个输入。

返回:

Tensor

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> # case 1 : same data type
>>> input = mindspore.tensor([1.0, 5.0, 3.0], mindspore.float32)
>>> other = mindspore.tensor([4.0, 2.0, 6.0], mindspore.float32)
>>> mindspore.ops.maximum(input, other)
Tensor(shape=[3], dtype=Float32, value= [ 4.00000000e+00,  5.00000000e+00,  6.00000000e+00])
>>>
>>> # case 2 : the data type is the one with higher precision or higher digits among the two inputs.
>>> input = mindspore.tensor([1.0, 5.0, 3.0], mindspore.int64)
>>> other = mindspore.tensor([4.0, 2.0, 6.0], mindspore.float64)
>>> mindspore.ops.maximum(input, other)
Tensor(shape=[3], dtype=Float64, value= [ 4.00000000e+00,  5.00000000e+00,  6.00000000e+00])