TensorFlow:我如何总结tf.Variables列表？

Question

我有一个tf.Variable类型的3D数组.tf.reduce_sum仅适用于个人张量.我试过做:

tf.reduce_sum([tf.reduce_sum(mat) for mat in var_3Dlist])

...但是tf.reduce_sum()期望张量而不是列表.我能以某种方式将其转换为张量,还是有另一种更好的方法呢？我没有在API中找到任何内容.

Answer 1

的tf.reduce_sum()运算工作在3-d张量和变量(和一般的任何等级或张量或变量).但是,如果你有一个清单的2-d张量(或变量),你应该使用tf.add_n()op将值相加列表:

var_3Dlist = ...  # List of 3-D variables.
sum_list = [tf.reduce_sum(mat) for mat in var_3Dlist]
sum = tf.add_n(sum_list)