如果我有矩阵
A = tf.fill([2,2],2)
然后是矩阵数组
B = tf.fill([3,2,2],2)
如何将矩阵A添加到B中的所有矩阵中以产生
[
[
[4,4],
[4,4]
],
[
[4,4],
[4,4]
],
[
[4,4],
[4,4]
]
]
那我该如何求和B中的每个矩阵以产生像这样的向量
[16,16,16]
?
我是Tensorflow的新手,到目前为止,我的经验使我感到用几行代码来完成此操作是一种优雅的方法。
答案 0 :(得分:1)
这里:
A = tf.fill([2,2],2)
B = tf.fill([3,2,2],2)
A_B = tf.add(A, B)
sum_A_B = tf.reduce_sum(tf.reduce_sum(A_B, axis=1), axis=1)