不同等级的张量批量tf.matmul

Question

给定一个形状(A, B, C, D, E)的张量和一个形状(A, B, E)的张量，我想对第二张量进行自动广播以进行批量乘法，使得：

    In [1]: X = tf.placeholder(dtype=tf.float32, shape=[A, B, C, D, E])
    In [2]: Y = tf.placeholder(dtype=tf.float32, shape=[A, B, E])
    In [3]: # DO SOMETHING LIKE tf.matmul(X, Y)
    Out[3]: <tf.Tensor 'MatMul_1:0' shape=(A, B, C, D) dtype=float64>

可能吗？

提前致谢

Answer 1

对于那些使用逐元素计算的操作，大多数情况下都支持广播。 tf.matmul不是基于元素的操作，而是tf.multiply是基于元素的。

同样，即使在支持广播的操作中，Tensorflow可能也可能不会针对高阶张量进行自动广播。 在这种情况下，您将必须手动执行此操作。

我写了另一个答案 ，它会自动广播高阶张量。