TensorDataset¶

class paddle.io. TensorDataset ¶

由 Tensor 列表定义的数据集。

每个 Tensor 的形状应为[N，...]，而 N 是样本数，每个 Tensor 表示样本中一个字段，TensorDataset 中通过在第一维索引 Tensor 来获取每个样本。