标签: tensorflow
假设我有维度A的矩阵(N,5)和维度v的向量(N,1)。有没有有效的方法来获得维度B的新矩阵(N,5),使得张量流中的每一行B[i,:]=v[i]*A[i,:]?
A
(N,5)
v
(N,1)
B
B[i,:]=v[i]*A[i,:]
已知操作tf.matmul和tf.multiply不支持这类操作。
tf.matmul
tf.multiply