.. index:: pair: page; LogSoftmax .. _doxid-dev_guide_op_logsoftmax: LogSoftmax ========== General ~~~~~~~ LogSoftmax operation applies the :math:`\log(softmax(src))` function to an n-dimensional input Tensor. The formulation can be simplified as: .. math:: dst_i = \log\Big( \frac{exp(src_i)}{\sum_{j}^{ } exp(src_j)} \Big) Operation attributes ~~~~~~~~~~~~~~~~~~~~ ================================================================================================================= =================================================================================================================== =========== ========================================= ===================== Attribute Name Description Value Type Supported Values Required or Optional ================================================================================================================= =================================================================================================================== =========== ========================================= ===================== :ref:`axis ` 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 ``src`` Required ====== ============== ===================== Outputs ------- ====== ============== ===================== Index Argument Name Required or Optional ====== ============== ===================== 0 ``dst`` Required ====== ============== ===================== Supported data types ~~~~~~~~~~~~~~~~~~~~ LogSoftmax operation supports the following data type combinations. ===== ===== Src Dst ===== ===== f32 f32 bf16 bf16 f16 f16 ===== ===== Implementation Notes ~~~~~~~~~~~~~~~~~~~~ LogSoftmax supports in-place operations, meaning that ``src`` can be used as both input and output (``dst``). In case of in-place operation, the original ``src`` data will be overwritten. Use in-place operations whenever possible for performance.