mindspore.dataset.dataloader.TensorDataset

class mindspore.dataset.dataloader.TensorDataset(*tensors: Tensor)[源代码]

Each sample is retrieved by indexing the input tensors along their first dimension.

Parameters: *tensors (mindspore.Tensor) – Input tensors. All tensors must have the same size in the first dimension.