triton.language.cumprod

triton.language.cumprod(input, axis=0, reverse=False, dtype: constexpr = None)

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

• dtype (tl.dtype) – the desired data type of the returned tensor. If specified, the input tensor is casted to dtype before the operation is performed. If not specified, signed integer dtypes narrower than 32 bits are upcasted to tl.int32, while unsigned integer and bool dtypes narrower than 32 bits are upcasted to tl.uint32. Other dtypes are kept as-is.

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