tenzr::data::Dataset¶

Abstract indexed collection of tensor samples.

#include <dataset.h>

Inherited by tenzr::data::ShardedDataset, tenzr::data::TensorDataset

Public Functions¶

Public Functions Documentation¶

function ~Dataset¶

virtual ~Dataset() =default

function Size¶

virtual size_t Size() const =0

Return: Number of samples.

Reimplemented by: tenzr::data::TensorDataset::Size, tenzr::data::ShardedDataset::Size

function Get¶

virtual DataItem Get(
    size_t index
) const =0

Parameters:

• index Zero-based sample index.

Return: Tensors making up the sample.

Reimplemented by: tenzr::data::TensorDataset::Get, tenzr::data::ShardedDataset::Get

Fetch one sample.