Files
openvino/docs/ops/activation/SoftMax_8.md
T
Maciej Smyk 0148076ed7 [DOCS] Code block update for master (#18437)
* code-block-1

* Update Convert_Model_From_Paddle.md

* code-block force

* fix

* fix-2

* Update troubleshooting-steps.md

* code-block-2

* Update README.md
2023-07-11 10:43:54 +02:00

1.4 KiB

SoftMax

@sphinxdirective

.. meta:: :description: Learn about SoftMax-8 - an element-wise, activation operation, which can be performed on a single tensor in OpenVINO.

Versioned name: SoftMax-8

Category: Activation function

Short description: Reference <https://github.com/Kulbear/deep-learning-nano-foundation/wiki/ReLU-and-Softmax-Activation-Functions#softmax>__

Detailed description: Reference <http://cs231n.github.io/linear-classify/#softmax>__

Attributes

  • axis

    • Description: axis represents the axis of which the SoftMax is calculated. Negative value means counting dimensions from the back. axis equal 1 is a default value.
    • Range of values: [-rank, rank - 1]
    • Type: int
    • Default value: 1
    • Required: no

Mathematical Formulation

.. math::

y_{c} = \frac{e^{Z_{c}}}{\sum_{d=1}^{C}e^{Z_{d}}}

where :math:C is a size of tensor along axis dimension.

Inputs:

  • 1: Input tensor with enough number of dimension to be compatible with axis attribute. Required.

Outputs:

  • 1: The resulting tensor of the same shape and type as input tensor.

Example

.. code-block:: xml :force:

<layer ... type="SoftMax" ... > ... ...

@endsphinxdirective