torch.utils.data.dataset.Subset Class Reference

Inheritance diagram for torch.utils.data.dataset.Subset:

Public Member Functions
def	__init__ (self, dataset, indices)
def	__getitem__ (self, idx)
def	__len__ (self)
Public Member Functions inherited from torch.utils.data.dataset.Dataset
def	__getitem__ (self, index)
def	__len__ (self)
def	__add__ (self, other)

Public Attributes
	dataset
	indices

Detailed Description

Subset of a dataset at specified indices.

Arguments:
    dataset (Dataset): The whole Dataset
    indices (sequence): Indices in the whole set selected for subset

Definition at line 94 of file dataset.py.

---

The documentation for this class was generated from the following file:

• torch/utils/data/dataset.py