triton.language.cumprod

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

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

参数:

• input (Tensor) – 输入值

• axis (int) – 执行扫描操作的维度

• reverse (bool) – 若为 true，则以相反方向执行扫描

此函数也可作为 tensor 的成员函数调用，即使用 x.cumprod(...) 代替 cumprod(x, ...)。