# Subset

Source: https://www.tensorplay.cn/docs/generated/tensorplay.utils.data.Subset.html

# Subset

class tensorplay.utils.data.Subset(dataset: [Dataset](tensorplay.utils.data.Dataset.html#tensorplay.utils.data.Dataset)[_T_co], indices: [Sequence](https://docs.python.org/3/library/collections.abc.html#collections.abc.Sequence)[[int](https://docs.python.org/3/library/functions.html#int)])[[source]](../_modules/tensorplay/utils/data.html#Subset)

Subset of a dataset at specified indices.

Note

When subclassing Subset and overriding __getitem__, you must also
override __getitems__ to ensure DataLoader works correctly with your
custom logic. If you override only __getitem__, a NotImplementedError
will be raised when using DataLoader.

Parameters:

- dataset ([Dataset](tensorplay.utils.data.Dataset.html#tensorplay.utils.data.Dataset)) – The whole Dataset

- indices ( sequence ) – Indices in the whole set selected for subset