triton.language.cumprod

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

返回input张量中所有元素沿指定axis的累积乘积

Parameters:

• input (Tensor) – 输入值

• axis (int) – 指定扫描操作应沿哪个维度进行

该函数也可以作为成员函数在tensor上调用， 使用x.cumprod(...)而不是 cumprod(x, ...)。