将矩阵添加到矩阵数组

时间:2019-02-02 12:00:52

标签: python tensorflow

如果我有矩阵

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的新手,到目前为止,我的经验使我感到用几行代码来完成此操作是一种优雅的方法。

1 个答案:

答案 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)