triton.language.cumsum

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

返回input张量中所有元素沿指定axis轴的累加和

Parameters:

• input (Tensor) – 输入值

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

此函数也可以作为成员函数在tensor上调用， 形式为x.cumsum(...)而非 cumsum(x, ...)。