Python – tensorflow.math.reduce_sum()

TensorFlow是谷歌设计的开源Python库，用于开发机器学习模型和深度学习神经网络。

reduce_sum()用于查找张量的各维元素之和。

语法: tensorflow.math.reduce_sum( input_tensor, axis, keepdims, name)

参数:

• input_tensor:它是要减少的数字张量。
• axis（可选）:它代表要减少的维度。它的值应该在[-rank(input_tensor), rank(input_tensor)]范围内。如果没有给这个值，所有的维度都会被减少。
• keepdims（可选）:它的默认值是False。如果它被设置为True，它将保留长度为1的缩小尺寸。
• name（可选）:它定义了该操作的名称。

返回：它返回一个张量。

示例 1:

# importing the library
import tensorflow as tf
 
# Initializing the input tensor
a = tf.constant([1, 2, 3, 4], dtype = tf.float64)
 
# Printing the input tensor
print('Input: ', a)
 
# Calculating result
res = tf.math.reduce_sum(a)
 
# Printing the result
print('Result: ', res)

输出:

Input:  tf.Tensor([1. 2. 3. 4.], shape=(4, ), dtype=float64)
Result:  tf.Tensor(10., shape=(), dtype=float64)

示例 2:

# importing the library
import tensorflow as tf
 
# Initializing the input tensor
a = tf.constant([[1, 2], [3, 4]], dtype = tf.float64)
 
# Printing the input tensor
print('Input: ', a)
 
# Calculating result
res = tf.math.reduce_sum(a, axis = 1, keepdims = True)
 
# Printing the result
print('Result: ', res)

输出:

Input:  tf.Tensor(
[[1. 2.]
 [3. 4.]], shape=(2, 2), dtype=float64)
Result:  tf.Tensor(
[[3.]
 [7.]], shape=(2, 1), dtype=float64)