Tensor Data Manager

Module for the Tensor-Data Manager class.

class _TensorDataManager(**kwargs)

Bases: _DataManagerInterface

Data manager for tensor-based data. It handles inputs stored as torch.Tensor or LabelTensor.

Initialization of the _TensorDataManager class.

Parameters:

kwargs (dict) – The keyword arguments for the tensor data manager.

to_batch()

Create a batch from the current tensor data manager.

Returns:

A new instance of _BatchManager with batched data.

Return type:

_BatchManager

static create_batch(items)

Create a batch from a list of _TensorDataManager items.

Parameters:

items (list[_TensorDataManager]) – A list of _TensorDataManager items to batch.

Returns:

A new instance of _BatchManager containing the batched data.

Return type:

_BatchManager