torch.utils.data.dataset.TensorDataset Class Reference

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

Public Member Functions
def	__init__ (self, tensors)
def	__getitem__ (self, index)
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
	tensors

Detailed Description

Dataset wrapping tensors.

Each sample will be retrieved by indexing tensors along the first dimension.

Arguments:
    *tensors (Tensor): tensors that have the same size of the first dimension.

Definition at line 26 of file dataset.py.

---

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

• torch/utils/data/dataset.py