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

问题描述

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

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

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

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

编辑：我不使用急切执行

解决方法

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,[2,[[9,[3,0]]])>