Function Description
Operator Function
Performs the Softmax function operation along a data dimension.
Formula

Parameter |
Description |
|---|---|
|
Input tensor. |
dst |
Output tensor. |
|
Dimension of the softmax operation. |
|
The outermost dimension. |
|
The innermost dimension. |
|
Coefficient used to generate numerically stable output results. The coefficient is calculated using the following formula, where ic is all intermediate dimensions from the outermost to the innermost dimension.
|
Parent topic: Softmax





