triton.language.softmax

triton.language.softmax = JITFunction(triton.triton_patch.language.standard:softmax)

Computes the element-wise softmax of x.

Parameters:

x (Block) – the input values

This function can also be called as a member function on tensor, as x.softmax(...) instead of softmax(x, ...).