Shortcuts

README Tutos

Simplifying PyTorch Memory Management with TensorDict

Simplifying PyTorch Memory Management with TensorDict

Slicing, Indexing, and Masking

Slicing, Indexing, and Masking

Pre-allocating memory with TensorDict

Pre-allocating memory with TensorDict

Functional Programming with TensorDict

Functional Programming with TensorDict

Exporting tensordict modules

Exporting tensordict modules

Manipulating the keys of a TensorDict

Manipulating the keys of a TensorDict

Building tensordicts from streams

Building tensordicts from streams

TensorDictModule

TensorDictModule

Manipulating the shape of a TensorDict

Manipulating the shape of a TensorDict

Using tensorclasses for datasets

Using tensorclasses for datasets

Using TensorDict for datasets

Using TensorDict for datasets

Benchmarking TensorDict serialization speed

Benchmarking TensorDict serialization speed

Batched data loading with tensorclasses

Batched data loading with tensorclasses

Gallery generated by Sphinx-Gallery

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources