fast_tensor_dataset

A special type of tensor data set for improved performance.

class FastTensorDataset(*args: Any, **kwargs: Any)[source]

Bases: Dataset

A special type of tensor data set for improved performance.

This version of TensorDataset gathers data using a single call within __getitem__. A bit more tricky to manage but is faster still.

shuffle()[source]

Shuffle the data set.