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.