github_url: | https://github.com/pytorch/pytorch |
---|
PyTorch is an optimized tensor library for deep learning using GPUs and CPUs.
.. toctree::
:glob:
:maxdepth: 1
:caption: Notes
notes/*
.. toctree::
:maxdepth: 1
:caption: Package Reference
torch
tensors
sparse
storage
nn
optim
torch.autograd <autograd>
torch.multiprocessing <multiprocessing>
torch.legacy <legacy>
cuda
ffi
data
model_zoo
.. toctree::
:glob:
:maxdepth: 1
:caption: torchvision Reference
torchvision/torchvision
torchvision/datasets
torchvision/models
torchvision/transforms
torchvision/utils