dataset_from_tensors

lightkit.data.dataset_from_tensors(*data)[source]

Transforms a set of tensor-like items into a datasets.

Parameters

data (Union[ndarray[Any, dtype[float32]], Tensor]) -- The tensor-like items.

Return type

TensorDataset

Returns

The dataset.