Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
torch.utils.data.dataset.Subset Class Reference
Inheritance diagram for torch.utils.data.dataset.Subset:
torch.utils.data.dataset.Dataset

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: