Rate This Document
Findability
Accuracy
Completeness
Readability

Function Description

Operator Function

Performs the Softmax function operation along a data dimension.

Formula

Table 1 Formula parameters

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.