SoftMax
SoftMax#
Versioned name: SoftMax-1
Category: Activation
Short description: Reference
OpenVINO description: This OP is as same as OpenVINO OP
Detailed description: Reference
Attributes:
axis
Description: axis represents the axis of which the SoftMax is calculated. axis equal 1 is a default value.
Range of values: [-r, r-1] where r = rank(input)
Type: s64
Default value: 1
Required: no
Inputs:
1: Input tensor with enough number of dimension to be compatible with axis attribute. Required.
Type: T
Outputs
1: The resulting tensor of the same shape as input tensor.
Type: T
Types:
T: f32, f16, bf16.
Note: Inputs and outputs have the same data type denoted by T. For example, if input is f32 tensor, then all other tensors have f32 data type.
Detailed description
where \(C\) is a size of tensor along axis dimension.