triton.softmax

triton.softmax(x)

Computes the element-wise softmax of x.

Parameters

x (Block) – the input block