如何在张量流中将一批矩阵乘以相应的张量（1 D）

Question

假设我有形状为 (3,2,2) 的张量，我想将每个 (2,2) 矩阵乘以某个 1d 张量。这是代码：

a=tf.constant([[[1,0],[1,0]], [[2,0],[1,0]], [[3,0],[1,0]] ])
b=tf.constant([1,2,3])

在这里，我想将[[1,0],[1,0]] 乘以 1，[[2,0],[1,0]] 乘以 2，[[3,0],[1,0]] 乘以 3。所以乘法后我应该得到：

[[[1,0],[1,0]], [[4,0],[2,0]], [[9,0],[3,0]] ]

编辑：我没有使用急切执行

Answer 1

tf.convert_to_tensor([tf.multiply(aa, bb) for aa, bb in zip(a, b)])

<tf.Tensor: shape=(3, 2, 2), dtype=int32, numpy=
array([[[1, 0],
        [1, 0]],
       [[4, 0],
        [2, 0]],
       [[9, 0],
        [3, 0]]])>