Tensorflow 中的平均歐幾里德距離

Question

我有兩個大小為 [batch_size, seq_length, 2] 的序列張量。 我想計算張量之間的平均歐幾里得距離。 這樣做的優雅方法是什么？

Answer 1

給定兩個張量A和B每個張量A形狀為[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
)

請參閱此處的文檔。