tensorflow 怎么对一个tensor进行所有元素相加求和

版权声明:本文系作者原创,未经允许,禁止转载。 https://blog.csdn.net/qq_27261889/article/details/85001486

tensor 和 numpy不同。
使用np.sum对一个tensor求和是不能得到一个标量的。

所以需要使用tensorflow自带的函数对tensor进行求和。
总的来说,这里我们需要明白一点,tensor数据使用tensorflow的函数运算,否则会出错。

这里仅给出对一个tensor进行所有元素相加求和的运算:
tf.reduce_sum(input)
这一行命令可以对input这个tensor进行元素相加。但是得到的仍然是tensor,并且也需要保证input是一个tensor

猜你喜欢

转载自blog.csdn.net/qq_27261889/article/details/85001486
今日推荐