Class RandomSelectSubpolicy

Inheritance Relationships

Base Type

Class Documentation

class RandomSelectSubpolicy : public mindspore::dataset::TensorTransform

Choose a random sub-policy from a list to be applied on the input image. A sub-policy is a list of tuples (operation, prob), where operation is a TensorTransform operation and prob is the probability that this operation will be applied. Once a sub-policy is selected, each operation within the sub-policy with be applied in sequence according to its probability.

Public Functions

explicit RandomSelectSubpolicy(const std::vector<std::vector<std::pair<TensorTransform*, double>>> &policy)

Constructor.

Parameters

policy – [in] Vector of sub-policies to choose from, in which the TensorTransform objects are raw pointers.

explicit RandomSelectSubpolicy(const std::vector<std::vector<std::pair<std::shared_ptr<TensorTransform>, double>>> &policy)

Constructor.

Parameters

policy – [in] Vector of sub-policies to choose from, in which the TensorTransform objects are shared pointers.

explicit RandomSelectSubpolicy(const std::vector<std::vector<std::pair<std::reference_wrapper<TensorTransform>, double>>> &policy)

Constructor.

Parameters

policy – [in] Vector of sub-policies to choose from, in which the TensorTransform objects are object pointers.

~RandomSelectSubpolicy() = default

Destructor.