triton.language.min

triton.language.min(input, axis)

Returns the minimum of all elements in the input block along the provided axis

Parameters
  • input – the input values

  • axis – the dimension along which the reduction should be done