TensorLoader Class Reference

#include <TensorLoader.h>

Inheritance diagram for TensorLoader:

Public Member Functions
	TensorLoader (const Tensor &tensor, Size batch_dim)
std::size_t	total () const override
	Total (fixed) number of batches that will be generated.
Public Member Functions inherited from FixedSizeWorkGenerator< Tensor >
bool	has_more () const override
	Whether the generator has more work to generate.
Public Member Functions inherited from WorkGenerator< Tensor >
	WorkGenerator ()=default
	WorkGenerator (WorkGenerator &&) noexcept=default
	WorkGenerator (const WorkGenerator &)=default
WorkGenerator &	operator= (WorkGenerator &&) noexcept=default
WorkGenerator &	operator= (const WorkGenerator &)=default
virtual	~WorkGenerator ()=default
std::pair< std::size_t, Tensor >	next (std::size_t n)
	Generate the next n batches of work.
std::size_t	offset () const
	Return the current offset, i.e., the number of batches that have been generated.

Protected Member Functions
std::pair< std::size_t, Tensor >	generate (std::size_t n) override
	Generate the next n batches of work.

Constructor & Destructor Documentation

TensorLoader()

TensorLoader	(	const Tensor &	tensor,
		Size	batch_dim )

Member Function Documentation

generate()

std::pair< std::size_t, Tensor > generate ( std::size_t n )

overrideprotectedvirtual

Generate the next n batches of work.

Note that in the case of insufficient remaining work, it is possible that the number of batches generated is less than n.

Parameters

n	Number of batches to generate

Returns

std::pair<std::size_t, T> Number of batches generated (m) and the next m batches of work

Implements WorkGenerator< Tensor >.

total()

std::size_t total ( ) const

overridevirtual

Total (fixed) number of batches that will be generated.

Implements FixedSizeWorkGenerator< Tensor >.