megengine.functional.minimum

minimum(x, y)[源代码]

Tensor中的逐元素最小值。

逐元素比较两个 tensor 的最小值,返回新 tensor。如果与”NaN”进行比较,则返回该元素。如果用于比较的两个元素都是”NaNs”,则返回第一个元素。

参数
  • x – 输入张量。 类型为数值数据类型。

  • y – 输入tensor。与 \(x\) 的数据类型一致。

返回

包含最小元素的tensor。返回tensor的数据类型必须与:math:`x`的数据类型一致。

实际案例

>>> F.minimum(1, 2)
Tensor(1, dtype=int32, device=xpux:0)

逐元素求最小值:

>>> x = Tensor([1, 2, 3, 4])
>>> y = Tensor([4, 3, 2, 1])
>>> F.minimum(x, y)
Tensor([1 2 2 1], dtype=int32, device=xpux:0)

广播:

>>> x = Tensor([1, 2, 3, 4])
>>> F.minimum(x, 2)
Tensor([1 2 2 2], dtype=int32, device=xpux:0)