triton.language.softmax

triton.language.softmax(x, ieee_rounding=False)

计算x的逐元素softmax。

Parameters:

x (Block) – 输入值

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