入门
Python API
cumprod()
Triton MLIR 方言
编程指南
返回 input 张量中所有元素沿提供的 axis 的累积乘积
input
axis
input (Tensor) – 输入值
axis (int) – 进行扫描操作的维度
此函数也可以作为成员函数在 tensor 上调用,例如 x.cumprod(...) 而不是 cumprod(x, ...)。
tensor
x.cumprod(...)
cumprod(x, ...)