Tensorflow中的平均欧几里德距离

Question

我有两个大小序列的张量[batch_size，seq_length，2]。我想计算张量之间的平均欧几里德距离。这样做的优雅方式是什么？

Answer 1

考虑到两个张量A＆amp;每个B形状为[batch_size, seq_length, 2]，您可以使用tf.norm计算欧几里德距离（L2范数）：

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

Answer 2

您也可以使用 tf.math.reduce_euclidean_norm：

tf.math.reduce_euclidean_norm(
    input_tensor, axis=None, keepdims=False, name=None
)

请参阅文档 here。