tensorflow-map_fn对两个张量的每个可能组合进行计算

有人知道如何使用map_fn或任何其他tensorflow func对两个输入张量的每个组合进行计算吗

所以我想要的是这样的： 由于有两个数组（

[1,2]

和

[4,5]

），因此，我希望在两个数组的每个可能组合上都有一个带有计算输出的矩阵（例如

add

）。因此，结果将是：

[[5,6],
 [6,7]]

我使用了map_fn，但这仅采用元素索引：

[[5]
 [7]]

有人知道如何实现这一点吗

---

多亏了

您可以为每个张量添加新的单位维度，然后依靠广播添加：

import tensorflow as tf
import tensorflow.contrib.eager as tfe
tfe.enable_eager_execution()
first = tf.constant([1, 2])
second = tf.constant([4, 5])
print(first[None, :] + second[:, None])

印刷品：

tf.Tensor(
[[5 6]
 [6 7]], shape=(2, 2), dtype=int32)