Linux 拨号vps windows公众号手机端

tensorflow中sum的用法是什么

lewis 5年前 (2020-02-03) 阅读数 10 #大数据
文章标签 tensorflow

在TensorFlow中,tf.reduce_sum函数用于计算张量中所有元素的总和。例如:

import tensorflow as tf

x = tf.constant([[1, 2, 3], [4, 5, 6]])
sum_x = tf.reduce_sum(x)

with tf.Session() as sess:
    result = sess.run(sum_x)
    print(result)

这将输出:

21

上面的代码将张量x中的所有元素相加,得到总和为21。

版权声明

本文仅代表作者观点,不代表米安网络立场。

发表评论:

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

热门