Tensorflow reduce_sum

Computes the sum of elements across dimensions of a tensor. 

x = tf.constant([[1, 1, 1], [1, 1, 1]])
tf.reduce_sum(x)  # 6
tf.reduce_sum(x, 0)  # [2, 2, 2]
tf.reduce_sum(x, 1)  # [3, 3]
tf.reduce_sum(x, 1, keepdims=True)  # [[3], [3]]
tf.reduce_sum(x, [0, 1])  # 6

Source: tensorflow.org

Similar codes

1 vote149 views1 answer
Python pip install all dependencies
Python
-2 votes144 views2 answers
Diamond pattern in python
Python
0 vote140 views2 answers
SQLAlchemy Users to JSON code snippet
Python
12 votes141 views2 answers
Max of a dict
Python
2 votes153 views1 answer
Filter by row contains pandas
Python
0 vote139 views2 answers
How to return a missing element in python
Python
8 votes143 views3 answers
Python execute function from string
Python
0 vote173 views1 answer
Python Key Gen
Python
6 votes174 views1 answer
How to print a float with only 2 digits after decimal in python
Python
1 vote146 views6 answers
Insertion sort
Python