0
在张量流中,我有以下问题。张量流中的广播点积
我有一个张量米形状的 [batch_size时,dim_a,dim_b]和基质ü形状[batch_size时,dim_b]的。
M = tf.constant(shape=[batch_size, sequence_size, embed_dim])
U = tf.constant(shape=[batch_size, embed_dim])
我该怎么实现的是[我,dim_a,dim_b]×[我,dim_b]我批各指标的点积。
P[i] = tf.matmul(M[i, :, :], tf.expand_dims(U[i, :], 1)) for each i.
基本上,broadacast在批处理轴上的点积。这是可能的,我该如何实现它?
由于每个载体与1列的矩阵,则可以重塑U. Tensorflow用于具有'在版本1.0之前的batch_matmul',现在已经与'matmul'合并了。 – dseuss