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() op适用于三维张量和变量（通常是任何等级或张量或变量）。但是，如果您有2-D张量（或变量）的列表，则应使用tf.add_n()操作将列表中的值相加：

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