Python Tensorflow：如何对矩阵的每个元素执行操作

我有两个矩阵A（Nx100）和B（Mx100）。我想对矩阵A中的每个元素和矩阵B中的每个元素执行一个运算（比如计算欧氏距离）。在正常的numpy中，这将使用两个for循环来完成

现在，我正在使用map函数

tf.map\u fn

将矩阵a中的每个元素映射到我执行的操作

a[I]-B

，其中a[I]是一个向量。这是因为TensorFlow将向量广播到矩阵的形状中

我想知道是否有一种方法可以让我更容易做到这一点。类似于：

tf.do_magic（A，B，operation）

---

谢谢

A和B的欧几里德距离：

tf.sqrt(tf.reduce_sum(tf.square(tf.expand_dims(A, 1) - B)), 2))