class paddle.io. TensorDataset ( tensors ) [source]

Dataset defined by a list of tensors.

Each tensor should be in shape of [N, …], while N is the sample number, and ecah tensor contains a field of sample, TensorDataset retrieve each sample by indexing tensors in the 1st dimension.


tensors (list|tuple) – A list/tuple of tensors with same shape in the 1st dimension.


a Dataset instance wrapping tensors.

Return type



import numpy as np
import paddle
from paddle.io import TensorDataset

input_np = np.random.random([2, 3, 4]).astype('float32')
input = paddle.to_tensor(input_np)
label_np = np.random.random([2, 1]).astype('int32')
label = paddle.to_tensor(label_np)

dataset = TensorDataset([input, label])

for i in range(len(dataset)):
    input, label = dataset[i]
    print(input, label)