Softmax

Softmax  // Softmax operation

Description

Produces a tensor of the same element type and shape as arg, where the value at each coordinate of output is the expine of the value of the corresponding coordinate of arg divided by the sum of the expine of all coordinates of arg in the specified axes.

Inputs

Name

Element Type

Shape

arg

Any

Any

Attributes

Name

Description

axes

The axis positions (0-based) on which to calculate the softmax

Outputs

Name

Element Type

Shape

output

Same as arg

Same as arg

Mathematical Definition

\[\mathtt{output}_{i} = \frac{\exp(\mathtt{arg}_{i})}{\sum_{j} \exp(\mathtt{arg}_{j})}\]

C++ Interface

class Softmax : public ngraph::op::util::UnaryElementwiseArithmetic

Softmax operation.