Tensorflow源码分析--add_n()

Tensorflow源码分析–add_n()

标签(空格分隔): Tensorflow


这个和之前的add()函数很像,只是它可以通过列表的方式输入多个Tensor

import tensorflow as tf
x = tf.constant([1,3])
y = tf.constant([3,2])
sess = tf.Session()
print(sess.run(tf.add_n([x,y,x])))
>>> [5 8]

猜你喜欢

转载自blog.csdn.net/jiangzhenkang/article/details/80744456