Skip to main content

function

softmax

softmax(value: Symbol) -> Symbol

Computes the elementwise softmax of a symbolic tensor.

Creates a new op node to compute the elementwise softmax of a symbolic tensor and adds it to the graph, returning the symbolic result.

Args:

  • value (Symbol): The symbolic tensor to use as the input to the softmax computation.

Returns:

A new symbolic tensor value representing the output of the absolute value computation.

Raises:

If the symbol doesn't represent a tensor value.