Intel® oneAPI Deep Neural Network Developer Guide and Reference
A newer version of this document is available. Customers should click here to go to the newest version.
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
Attribute Name |
Description |
Value Type |
Supported Values |
Required or Optional |
---|---|---|---|---|
Represents the axis from which the SoftMax is calculated. |
s64 |
Arbitrary s64 value ( 1 in default) |
Optional |
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 |
bf16 |
f16 |
f16 |