Intel® oneAPI Deep Neural Network Developer Guide and Reference
SoftMax
General
SoftMax operation applies the following formula on every element of tensor (the variable names follow the standard Naming Conventions):

where is a size of tensor along axis dimension.
Operation attributes
When the operation attribute mode is not set or set to none, the operation performs the normal SoftMax calculation. In this case, the operation will generate NaN if all the input elements are -infinity along the axis dimension. To prevent this, you can set the attribute to inf_as_zero so that the operation generates zeros for -infinity inputs.
Execution arguments
The inputs and outputs must be provided according to below index order when constructing an operation.
Inputs
Index |
Argument Name |
Required or Optional |
---|---|---|
0 |
src |
Required |
Outputs
Index |
Argument Name |
Required or Optional |
---|---|---|
0 |
dst |
Required |
Supported data types
SoftMax operation supports the following data type combinations.
Src |
Dst |
---|---|
f32 |
f32, bf16, f16 |
bf16 |
bf16 |
f16 |
f16 |