我要评分
获取效率
正确性
完整性
易理解

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.