tfep.io.dataset

Modules

dict

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

merged

Utility class to merge multiple PyTorch Datasets.

traj

Utility classes to create PyTorch ``Dataset``s from MDAnalysis trajectories.