From aecc9741c09eccd098ce3e349925d61ffe6dd6d5 Mon Sep 17 00:00:00 2001
From: dengkaipeng <dengkaipeng@baidu.com>
Date: Sun, 24 Feb 2019 21:58:25 +0800
Subject: [PATCH] fix pool3d doc. test=develop

---
 python/paddle/fluid/layers/nn.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py
index 2315a2d5cc..51f927dba5 100644
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@@ -2522,6 +2522,7 @@ def pool2d(input,
     return pool_out
 
 
+@templatedoc()
 def pool3d(input,
            pool_size=-1,
            pool_type="max",
@@ -2537,7 +2538,11 @@ def pool3d(input,
     pooling configurations mentioned in input parameters.
 
     Args:
-        input (Variable): ${input_comment}
+        input (Variable): The input tensor of pooling operator. The format of
+                          input tensor is NCDHW, where N is batch size, C is
+                          the number of channels, D is the depth of the feature,
+                          H is the height of the feature, and W is the width
+                          of the feature.
         pool_size (int): ${ksize_comment}
         pool_type (str): ${pooling_type_comment}
         pool_stride (int): stride of the pooling layer.
-- 
GitLab