lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
JittorMirror/doc/source/jittor.nn.md

513 B
Raw Blame History

jittor.nn

这里是Jittor的神经网络模块的API文档，您可以通过from jittor import nn来获取该模块。

.. automodule:: jittor.nn
   :members:
   :undoc-members:

.. automodule:: jittor.nn
   :imported-members:
   :members: Pool, pool, AdaptiveAvgPool2d
   :undoc-members: 

.. autoclass:: jittor.nn.ReLU
   :members:
.. autoclass:: jittor.nn.ReLU6
   :members:
.. autoclass:: jittor.nn.LeakyReLU
   :members:
.. autoclass:: jittor.nn.Softmax
   :members: