triton.language.cumprod

triton.language.cumprod(input, axis=0)

Returns the cumprod of all elements in the input tensor along the provided axis

Parameters:
  • input – the input values

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