SoftmaxBackward¶

General¶

SoftmaxBackward operation computes gradient for Softmax.

Attribute Name	Descr
axis	Represents the axis of 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	Argu
0	`diff_dst`	Required
1	`src`	Required

Index	Argu
0	`diff_src`	Required

SoftmaxBackward operation supports the following data type combinations.

Softmax SoftPlus