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.
-
explicit MixUpBatch(float alpha = 1)