Class Softmax

Inheritance Relationships

Base Type

  • public mindspore::ops::BaseOperator

Class Documentation

class Softmax : public mindspore::ops::BaseOperator

Softmax operation. Refer to Python API mindspore.ops.Softmax for more details.

Public Functions

inline Softmax()

Constructor.

void Init(const int64_t axis = -1)

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

void set_axis(const std::vector<int64_t> &axis)

Set axis.

std::vector<int64_t> get_axis() const

Get axis.

Returns

axis.