extending.rst 256 Bytes
Newer Older
Hang Zhang's avatar
docs  
Hang Zhang committed
1
2
3
4
5
6
7
8
9
Extending PyTorch-Encoding
==========================

In this note we'll discuss extending PyTorch-Encoding package,
which is extending :mod:`torch.nn` and
:mod:`torch.autograd` with custom CUDA backend.

Torch C and CUDA Backend
------------------------