triton.language.min

triton.language.min(input, axis=None, return_indices=False, return_indices_tie_break_left=True, keep_dims=False)

返回 input 张量沿提供的 axis 的所有元素的最小值

参数:

• input (Tensor) – 输入值

• axis (int) – 进行归约的维度。如果为 None，则归约所有维度

• keep_dims (bool) – 如果为 True，则保持长度为 1 的归约维度

• return_indices (bool) – 如果为 True，则返回对应于最小值的索引

• return_indices_tie_break_left (bool) – 如果为 True，则在出现平局（即多个元素具有相同的最小值）时，返回非 NaN 值的最左侧索引

此函数也可以作为成员函数在 tensor 上调用，例如 x.min(...) 而不是 min(x, ...)。