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.