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'</