PyTorch torch.cdist函数介绍及示例代码

1. torch.cdist 函数介绍

torch.cdist 是 PyTorch 中用于计算两组向量之间成对距离的函数。它可以计算两个张量（矩阵）中的每对向量之间的距离，支持多种距离度量方式，如欧氏距离（默认）或 p 范数距离。

函数原型

torch.cdist(x1, x2, p=2.0, compute_mode='use_mm_for_euclid_dist_if_necessary')

参数说明

• x1：形状为 (batch_size, n, m) 的张量，表示第一组向量，其中 n 是向量的数量，m 是每个向量的维度。
• x2：形状为 (batch_size, k, m) 的张量，表示第二组向量，其中 k 是向量的数量，m 是每个向量的维度。
• p：指定距离的类型，默认为 2，即计算欧氏距离。也可以指定为其他的 p 值来计算 p 范数距离。
• compute_mode（可选）：指定计算模式，可以是'use_mm_for_euclid_dist_if_necessary'</