Class OneHot
Defined in File one_hot.h
Inheritance Relationships
Base Type
public mindspore::ops::PrimitiveC
(Class PrimitiveC)
Class Documentation
-
class OneHot : public mindspore::ops::PrimitiveC
Computes a one-hot tensor. Refer to Python API mindspore.ops.OneHot for more details.
Public Functions
-
inline OneHot()
Constructor.
-
~OneHot() = default
Destructor.
-
void Init(const int64_t axis)
Init. Refer to the parameters of Python API mindspore.ops.OneHot for the inputs.
-
void set_axis(const int64_t axis)
Set axis.
-
int64_t get_axis() const
Get axis.
- Returns
axis.
-
inline OneHot()