triton.language.cumprod

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

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

Parameters:

• input (Tensor) – the input values

• axis (int) – the dimension along which the scan should be done

• reverse (bool) – if true, the scan is performed in the reverse direction

This function can also be called as a member function on tensor, as x.cumprod(...) instead of cumprod(x, ...).