用于张量乘法的 keras/tensorflow 广播答案

【问题标题】：keras/tensorflow broadcast for tensor multiplication用于张量乘法的 keras/tensorflow 广播
【发布时间】：2020-06-12 15:03:43
【问题描述】：

我有两个张量：张量a 的形状为(n, 1)，张量b 的形状为(p, q)，其中p 和q 都很小，但n 很大。我想将他们的“外积”张量 c 作为形状为 (n, p, q) 的张量，这样它的元素就是

c[i, j, k] = a[i, 0] * b[j, k]

我应该如何使用 tensorflow/keras 广播或函数来实现这一点而不会产生循环？

【问题讨论】：

【解决方案1】：

你可以这样做：

c = tf.expand_dims(a, 2) * b

【讨论】：