tfep.io.dataset.dict

Utility class to create a map-style PyTorch Datasets from a dictionary of tensors.

For usage examples see the documentation of DictDataset.

Classes

DictDataset(tensor_dict)

Utility class to create a map-style PyTorch Datasets from a dictionary of tensors.