TensorDataset

class tensorplay.utils.data.TensorDataset(*tensors: TensorBase)[source]

Dataset wrapping tensors.

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

Parameters:

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