Class MixUpBatch

Defined in File vision.h

Inheritance Relationships

Base Type

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

Class Documentation

class MixUpBatch : public mindspore::dataset::TensorTransform 

Apply MixUp transformation on an input batch of images and labels. The labels must be in one-hot format and Batch must be called before calling this function.

Public Functions

explicit MixUpBatch(float alpha = 1)

Constructor.

Parameters: alpha – [in] hyperparameter of beta distribution (default = 1.0).

~MixUpBatch() = default: Destructor.