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(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.