Intel® oneAPI Deep Neural Network Developer Guide and Reference
A newer version of this document is available. Customers should click here to go to the newest version.
LogSoftmaxBackward
General
LogSoftmaxBackward operation computes gradient for LogSoftmax.
Operation attributes
Attribute Name  |  
        Description  |  
        Value Type  |  
        Supported Values  |  
        Required or Optional  |  
       
|---|---|---|---|---|
Represents the axis of which the LogSoftmax is calculated. Negative value means counting dimensions from the back.  |  
        s64  |  
        Arbitrary s64 value ( -1 in default)  |  
        Optional  |  
       
Execution arguments
The inputs and outputs must be provided according to below index order when constructing an operation.
Inputs
Index  |  
        Argument Name  |  
        Required or Optional  |  
       
|---|---|---|
0  |  
        diff_dst  |  
        Required  |  
       
1  |  
        dst  |  
        Required  |  
       
Outputs
Index  |  
        Argument Name  |  
        Required or Optional  |  
       
|---|---|---|
0  |  
        diff_src  |  
        Required  |  
       
Supported data types
LogSoftmaxBackward operation supports the following data type combinations.
Diff_dst  |  
        Dst  |  
        Diff_src  |  
       
|---|---|---|
f32  |  
        f32  |  
        f32  |  
       
bf16  |  
        bf16  |  
        bf16  |  
       
f16  |  
        f16  |  
        f16  |