# TensorDataset

Source: https://www.tensorplay.cn/docs/generated/tensorplay.utils.data.TensorDataset.html

# TensorDataset

class tensorplay.utils.data.TensorDataset(*tensors: TensorBase)[[source]](../_modules/tensorplay/utils/data.html#TensorDataset)

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.