.. index:: pair: page; SoftmaxBackward
.. _doxid-dev_guide_op_softmaxbackward:

SoftmaxBackward
===============

General
~~~~~~~

SoftmaxBackward operation computes gradient for Softmax.

Operation attributes
~~~~~~~~~~~~~~~~~~~~

=================================================================================================================  ========================================================  ====  ========================================  =========  
Attribute Name                                                                                                     Descr                                                     
=================================================================================================================  ========================================================  ====  ========================================  =========  
:ref:`axis <doxid-classdnnl_1_1graph_1_1op_1ac7650c0c15849338f9c558f53ce82684a433169d5d9bcbb6d43f0d288e68f0cad>`   Represents the axis of which the Softmax is calculated.   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   Argu           
======  =============  =========  
0       ``diff_dst``   Required   
1       ``src``        Required   
======  =============  =========

Outputs
-------

======  =============  =========  
Index   Argu           
======  =============  =========  
0       ``diff_src``   Required   
======  =============  =========

Supported data types
~~~~~~~~~~~~~~~~~~~~

SoftmaxBackward operation supports the following data type combinations.

=====  =======  =====  
Src    Diff_d   
=====  =======  =====  
f32    f32      f32    
bf16   bf16     bf16   
f16    f16      f16    
=====  =======  =====