张量与批量数据和矩阵的乘法

Question

给定形状A的张量[?, n, m]和形状W的张量[m, m]，我希望将形状{{1}的每个张量a相乘带[n,m]的{​​{1}}导致形状张量A。

我认为我可以通过重塑W来塑造[?, n, m]，从而以某种方式做到这一点，但这不会导致形状张量W。

Answer 1

是的，您确实可以通过重塑来实现这一目标：

tf.reshape(tf.matmul(tf.reshape(A, [-1, m]), W), [-1, n, m])