python - Tensorflow 中的平均欧氏距离

Question

我有两个大小为 [batch_size, seq_length, 2] 的序列张量。我想计算张量之间的平均欧氏距离。执行此操作的优雅方法是什么？

Answer 1

给定两个张量 A 和 B，每个张量的形状都是 [batch_size, seq_length, 2]，您可以计算欧氏距离 (L2范数)使用 tf.norm :

l2_norm = tf.norm(A-B, ord='euclidean')