add attention doc

This commit is contained in:
Dun Liang 2021-06-24 11:18:31 +08:00
parent 8aa478fa5e
commit acd55b4f1b
2 changed files with 11 additions and 0 deletions

View File

@ -28,6 +28,7 @@
jittor.linalg
jittor.console
jittor.distributions
jittor.attention
.. toctree::

View File

@ -0,0 +1,10 @@
jittor.attention
=====================
这里是Jittor的 数据变换 模块的API文档您可以通过`from jittor import attention`来获取该模块。
```eval_rst
.. automodule:: jittor.attention
:members:
:undoc-members:
```