diff --git a/doc/source/jittor.nn.md b/doc/source/jittor.nn.md index 0ef2b09f..0e5cd35d 100644 --- a/doc/source/jittor.nn.md +++ b/doc/source/jittor.nn.md @@ -10,7 +10,7 @@ jittor.nn .. 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 + :members: Pool, pool, AdaptiveAvgPool2d, Pool3d, AdaptiveMaxPool2d, AdaptiveAvgPool3d, AdaptiveMaxPool3d, pool3d, AvgPool2d, AvgPool3d, avg_pool2d, MaxPool2d, MaxPool3d, max_pool2d, max_pool3d, MaxUnpool2d, MaxUnpool3d :undoc-members: .. autoclass:: jittor.nn.ReLU