不同等級的張量批量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可能也可能不會針對高階張量進行自動廣播。 在這種情況下，您將必須手動執行此操作。

我寫了另一個答案 ，它會自動廣播高階張量。