Class LogSoftmax

• Defined in File log_softmax.h

Inheritance Relationships

Base Type

• public mindspore::ops::BaseOperator

Class Documentation

class LogSoftmax : public mindspore::ops::BaseOperator

Log Softmax activation function. Refer to Python API mindspore.ops.LogSoftmax for more details.

Public Functions

inline LogSoftmax()

Constructor.

void Init(const int64_t axis = -1)

Init. Refer to the parameters of Python API mindspore.ops.LogSoftmax for the inputs.

void set_axis(const int64_t axis)

Set axis.

int64_t get_axis() const

Get axis.

Returns

axis.