Softmax¶

General¶

Softmax operation applies following formula on every element of $src$ tensor (the variable names follow the standard Naming Conventions):

d s t_{i} = \frac{e x p (s r c_{i})}{\sum_{j = 1}^{C} e x p (s r c_{j})}

where $C$ is a size of tensor along axis dimension.

Attribute Name	Descr
axis	Represents the axis of which the Softmax is calculated.	s64	Arbitrary s64 value ( `1` in default)	Optional

The inputs and outputs must be provided according to below index order when constructing an operation.

Index	Argu
0	`src`	Required

Index	Argu
0	`dst`	Required

Softmax operation supports the following data type combinations.

SigmoidBackward SoftmaxBackward