Class OneHot

• Defined in File transforms.h

Inheritance Relationships

Base Type

• public mindspore::dataset::TensorTransform (Class TensorTransform)

Class Documentation

class OneHot : public mindspore::dataset::TensorTransform

Convert the labels into OneHot format.

Public Functions

explicit OneHot(int32_t num_classes)

Constructor.

Parameters

num_classes – [in] number of classes.

~OneHot() = default

Destructor.