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

692 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, Pool3d, AdaptiveMaxPool2d, AdaptiveAvgPool3d, AdaptiveMaxPool2d, pool3d, AvgPool2d, AvgPool3d, avg_pool2d, MaxPool2d, MaxPool3d, max_pool2d, max_pool3d, MaxUnpool2d, MaxUnpool3d
   :undoc-members: 

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