PyTorch中cdist和sum函数使用示例详解

以下是PyTorch中cdist与sum函数的联合使用详解：

1. cdist函数解析

功能：计算两个张量间的成对距离矩阵
输入格式：

• X1：形状为(B, P, M)的张量
• X2：形状为(B, R, M)的张量
• p：距离类型（默认2表示欧式距离）

输出：形状为(B, P, R)的距离矩阵，其中元素$d_{ij}$表示$X1[i]$与$X2[j]$之间的距离

2. sum函数解析