SoftMaxBackward¶

General¶

SoftMaxBackward operation computes gradient for SoftMax.

Attribute Name	Description	Value Type	Supported Values	Required or Optional
axis	Represents the axis from which the SoftMax is calculated.	s64	Arbitrary s64 value ( `1` in default)	Optional

The inputs and outputs must be provided according to below index order when constructing an operation.

Index	Argument Name	Required or Optional
0	`diff_dst`	Required
1	`dst`	Required

Index	Argument Name	Required or Optional
0	`diff_src`	Required

SoftMaxBackward operation supports the following data type combinations.

SoftMax SoftPlus