softmax

systemds.operator.algorithm.softmax(S: Matrix)

Performs softmax on the given input matrix.

Parameters:

S – Inputs of shape (N, D).

Returns:

Outputs of shape (N, D).