目录

Utility Functions

DataPipe Graph Visualization

to_graph

Visualizes a DataPipe by returning a graphviz.Digraph, which is a graph of the data pipeline.

Common Utility Functions

pin_memory_fn

Utility function to move data to pinned memory.

File Object and Stream Utility

StreamWrapper

StreamWrapper is introduced to wrap file handler generated by DataPipe operation like FileOpener.

DataLoader

For documentation related to DataLoader, please refer to the torch.utils.data documentation. Or, more specifically, the DataLoader API section.

DataLoader v2 is currently in development. For more information please refer to DataLoader2.

Sampler

For documentation related to Sampler, please refer to the torch.utils.data documentation on Data Loading order. The Sampler API section is here.

文档

访问 PyTorch 的全面开发人员文档

查看文档

教程

获取面向初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并解答您的问题

查看资源