Class Softmax
Defined in File softmax.h
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.
-
inline Softmax()