.. index:: pair: struct; dnnl_softmax_desc_t .. _doxid-structdnnl__softmax__desc__t: struct dnnl_softmax_desc_t ========================== .. toctree:: :hidden: Overview ~~~~~~~~ A descriptor of a Softmax operation. :ref:`More...` .. ref-code-block:: cpp :class: doxyrest-overview-code-block #include struct dnnl_softmax_desc_t { // fields :ref:`dnnl_primitive_kind_t` :ref:`primitive_kind`; :ref:`dnnl_prop_kind_t` :ref:`prop_kind`; :ref:`dnnl_memory_desc_t` :ref:`data_desc`; :ref:`dnnl_memory_desc_t` :ref:`diff_desc`; int :ref:`softmax_axis`; }; .. _details-structdnnl__softmax__desc__t: Detailed Documentation ~~~~~~~~~~~~~~~~~~~~~~ A descriptor of a Softmax operation. Fields ------ .. index:: pair: variable; primitive_kind .. _doxid-structdnnl__softmax__desc__t_1aada503f0c7cd766680d6a8720993b108: .. ref-code-block:: cpp :class: doxyrest-title-code-block :ref:`dnnl_primitive_kind_t` primitive_kind The kind of primitive. Used for self-identifying the primitive descriptor. Must be :ref:`dnnl_softmax `. .. index:: pair: variable; prop_kind .. _doxid-structdnnl__softmax__desc__t_1a63ef3b7d75b3870d50cb209e0a265e47: .. ref-code-block:: cpp :class: doxyrest-title-code-block :ref:`dnnl_prop_kind_t` prop_kind The kind of propagation. Possible values: :ref:`dnnl_forward_training ` and :ref:`dnnl_forward_inference `. .. index:: pair: variable; data_desc .. _doxid-structdnnl__softmax__desc__t_1abf2142a8799e960f8cb14bd661739ea8: .. ref-code-block:: cpp :class: doxyrest-title-code-block :ref:`dnnl_memory_desc_t` data_desc Source and destination memory descriptor. .. index:: pair: variable; diff_desc .. _doxid-structdnnl__softmax__desc__t_1affbec0ec38fb9093bad14be278a11874: .. ref-code-block:: cpp :class: doxyrest-title-code-block :ref:`dnnl_memory_desc_t` diff_desc Source and Destination of gradient memory descriptor. .. index:: pair: variable; softmax_axis .. _doxid-structdnnl__softmax__desc__t_1ae2923dc4fc7bea2fe6bf0745de54bffd: .. ref-code-block:: cpp :class: doxyrest-title-code-block int softmax_axis The axis along which to perform the softmax.