frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Updated Module Onboarding Lab (markdown)

Joel Schlosser
2021-08-27 10:32:32 -04:00
parent 172e1c9fae
commit abd4d64df1
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
Module-Onboarding-Lab.md

@ -33,6 +33,7 @@ each of the wrappers above.
* `torch/nn/modules/linear.py` - add the Python module class definition here
* `torch/nn/modules/__init__.py` - add the module class name to the list in `__all__`
* `torch/nn/functional.py` - add the Python functional form of the module here
* `torch/csrc/api/include/torch/nn/functional/linear.h` - add the C++ functional form here
* `torch/csrc/api/include/torch/nn/modules/linear.h` / `torch/csrc/api/src/nn/modules/linear.cpp` - add the C++ module class definitions here
* `torch/csrc/api/include/torch/nn/options/linear.h` / `torch/csrc/api/src/nn/options/linear.cpp` - add options for the module here (i.e. `num_features`)
* `torch/testing/_internal/common_nn.py` - add a `dict` entry to the `module_tests` list defining how to generate generic tests for the new module (note: this part is in the process of being improved)