Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
35 changes: 27 additions & 8 deletions python/paddle/v2/fluid/layers/tensor.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,9 +43,28 @@ def concat(input, axis):


def sums(input, out=None):
"""
This function takes in the input and performs the sum operation on it
and returns that as the output.
"""This function performs the sum operation on the input and returns the
result as the output.

Args:
input (Variable|list): The input tensor that has the elements
that need to be summed up.

Returns:
Variable: The tensor type variable that has the sum of input
written to it.

Examples:
.. code-block::python

tmp = fluid.layers.zeros(shape=[10], dtype='int32')
i = fluid.layers.fill_constant(shape=[1], dtype='int64', value=10)
a0 = layers.array_read(array=tmp, i=i)
i = layers.increment(x=i)
a1 = layers.array_read(array=tmp, i=i)
mean_a0 = layers.mean(x=a0)
mean_a1 = layers.mean(x=a1)
a_sum = layers.sums(input=[mean_a0, mean_a1])
"""
helper = LayerHelper('sum', **locals())
if out is None:
Expand All @@ -68,9 +87,9 @@ def fill_constant(shape, dtype, value, out=None):
"""
**fill_constant**

This function creates a tensor of specified *shape* and
This function creates a tensor of specified *shape* and
*dtype*, and initializes this with a constant supplied in *value*.

It also sets *stop_gradient* to True.

Args:
Expand Down Expand Up @@ -110,9 +129,9 @@ def fill_constant_batch_size_like(input,
"""
**fill_constant_batch_size_like**

This function creates a tensor of specified *shape*, *dtype* and batch size,
and initializes this with a constant supplied in *value*. The batch size is
obtained from the `input` tensor.
This function creates a tensor of specified *shape*, *dtype* and batch size,
and initializes this with a constant supplied in *value*. The batch size is
obtained from the `input` tensor.

It also sets *stop_gradient* to True.

Expand Down