# LogSoftmax#

LogSoftmax operation applies the $$\log(softmax(src))$$ function to an n-dimensional input Tensor. The formulation can be simplified as:

$\dst_i = \log\Big( \frac{e^(\src_i)}{\sum_{j}^{ } e^(\src_j)} \Big)$

## Operation Attributes#

Attribute

Name

Description

Value Type

Supported

Values

Required or

Optional

axis

Represents the axis of which the LogSoftmax is calculated.

s64

Arbitrary s64 value (-1 in default)

Optional

## Execution Arguments#

The inputs and outputs must be provided according to the 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