From f00ffa38612e96cf3782f2897407095dd22b8197 Mon Sep 17 00:00:00 2001 From: Dun Liang Date: Mon, 5 Jul 2021 10:58:01 +0800 Subject: [PATCH] add unpool doc --- doc/source/jittor.nn.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/doc/source/jittor.nn.md b/doc/source/jittor.nn.md index 2e728f81..0ef2b09f 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 + :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