Visible to Intel only — GUID: GUID-F7A0C9EB-871F-4C9F-BFB7-9C19500E5955
Visible to Intel only — GUID: GUID-F7A0C9EB-871F-4C9F-BFB7-9C19500E5955
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 |