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, ...)。