gpt4 book ai didi

python - 如何在 Tensorflow 中进行列求和?

转载 作者:太空狗 更新时间:2023-10-29 17:01:37 24 4
gpt4 key购买 nike

Tensorflow 中以下内容的等价物是什么?

np.sum(A, axis=1)

最佳答案

tf.reduce_sum 是一个更强大的工具。

# 'x' is [[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, keep_dims=True) ==> [[3], [3]]
tf.reduce_sum(x, [0, 1]) ==> 6

关于python - 如何在 Tensorflow 中进行列求和?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/35689671/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com