triton.language.softmax

triton.language.softmax(x, ieee_rounding: triton.language.core.constexpr = False)

Computes the element-wise softmax of x

Parameters

x (Block) – the input values