The softmax function#

Aliases#

halerium.core.softmax
halerium.core.math.softmax
softmax(operand, axis=- 1)#

Softmax of elements along specified axis.

Parameters:

  • operand – The operand to apply softmax

  • axis (None, int) – The axis along which to apply softmax.

Returns:

operator – The halerium Operator representing the operation.

Return type:

halerium.core.operator.Softmax