Class TakeDataset

Inheritance Relationships

Base Type

Class Documentation

class TakeDataset : public mindspore::dataset::Dataset

The result of applying the Take operator to the input Dataset.

Public Functions

TakeDataset(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() = default

Destructor of TakeDataset.