LogSoftmaxBackward#

LogSoftmaxBackward operation computes gradient for LogSoftmax.

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

diff_dst

Required

1

dst

Required

Outputs#

Index

Argument Name

Required or Optional

0

diff_src

Required

Supported Data Types#

LogSoftmaxBackward operation supports the following data type combinations.

Diff_dst

Dst

Diff_src

f32

f32

f32

bf16

bf16

bf16

f16

f16

f16