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)