diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py index 90af75a24f89eaf1946c72a9069fc9fe06711205..69f0f8dc898b710256d3b2e9f72a5de94b81485c 100644 --- a/python/paddle/fluid/layers/nn.py +++ b/python/paddle/fluid/layers/nn.py @@ -7506,9 +7506,16 @@ def space_to_depth(x, blocksize, name=None): space_to_depth is used to This operation is useful for resizing the activations between convolutions (but keeping all data) + - Non-overlapping blocks of size block_size x block size are rearranged into depth at each location. + - The depth of the output tensor is block_size * block_size * input channel + - The Y, X coordinates within each block of the input become the high order component of the output channel index + - channel should be divisible by square of blocksize + - height, width should be divsible by blocksize + + Args: x(variable): The input LoDtensor. - blocksize(variable): The blocksize to select the element on each feature map + blocksize(variable): The blocksize to select the element on each feature map should be > 2 Returns: Variable: The output LoDtensor.