/torch-nested

Easily manipulate torch.Tensors inside highly nested data-structures.

Primary LanguagePythonMIT LicenseMIT

Watchers