Versioned name: SoftMax-8
Category: Activation function
Short description: Reference
Detailed description: Reference
Description: axis represents the axis of which the SoftMax is calculated. Negative value means counting dimensions from the back. axis equal 1 is a default value.
Range of values:
[-rank, rank - 1]
Default value: 1
where \(C\) is a size of tensor along axis dimension.
1: Input tensor with enough number of dimension to be compatible with axis attribute. Required.
1: The resulting tensor of the same shape and type as input tensor.
<layer ... type="SoftMax" ... > <data axis="1" /> <input> ... </input> <output> ... </output> </layer>