Class TakeDataset
Defined in File datasets.h
Inheritance Relationships
Base Type
public mindspore::dataset::Dataset(Class Dataset)
Class Documentation
-
class TakeDataset : public mindspore::dataset::Dataset
The result of applying the Take operator to the input Dataset.
Public Functions
-
TakeDataset(const std::shared_ptr<Dataset> &input, int32_t count)
Constructor of TakeDataset.
Note
Takes count elements in this dataset.
- Parameters
input – [in] The dataset which need to apply take operation.
count – [in] Number of elements the dataset to be taken.
-
~TakeDataset() override = default
Destructor of TakeDataset.
-
TakeDataset(const std::shared_ptr<Dataset> &input, int32_t count)