PyTorch 中的距离函数深度解析：掌握向量间的距离和相似度计算

目录

Pytorch中Distance functions详解

pairwise_distance

用途

用法

参数

数学理论公式

示例代码

cosine_similarity

用途

用法

参数

数学理论

示例代码 

输出结果

pdist

用途

用法

参数

数学理论

示例代码

总结 

---

Pytorch中Distance functions详解

pairwise_distance

torch.nn.functional.pairwise_distance 是 PyTorch 中的一个函数，用于计算两组向量之间的成对距离。这个函数广泛应用于机器学习和深度学习中，尤其是在处理距离相关的任务，如聚类、相似度计算等。

用途

• 计算两组向量间的成对距离，常用于度量向量间的相似性或差异性。
• 用于机器学习中的距离度量，如k-最近邻 (k-NN)、聚类等。

用法

torch.nn.functional.pairwise_distance(x1, x2, p=2.0, eps=1e-6, keepdim=False)

 

• x1, x2: 输入的两组向量，必须有相同的维度。
• p: 距离计算的幂指数，默认为2，即欧几里得距离。
• eps: 一个小的数值，用于保证数值稳定性。
• keepdim: 是否保持输出的维度。

参数

• x1: 第一组向量的张量。
• x2: 第二组向量的张量。
• p: 距离度量的幂指数，默认为2（欧几里得距离）。
• eps: 避免除零错误的小数，默认为1e-6。
• keepdim: 在输出中保持原始输入的维度结构。

数学理论公式

对于向量 ​ 和 ，pairwise_distance 计算的是 p 范数下的距离：

 其中， 和  分别是​ 和  ，的第j个元素。

示例代码

import torch
import torch.nn.functional as F

# 定义两个向量组
x1 = torch.tensor([[1, 2, 3], [4, 5, 6]], dtype=torch.float32)
x2 = torch.tensor([[1, 3, 5], [2, 4, 6]], dtype=torch.float32)

# 计算成对距离
dist = F.pairwise_distance(x1, x2, p=2)

# 输出结果  tensor([2.2361, 2.4495]) 这里，输出的是每一对向量之间的欧几里得距离。

print(dist)

cosine_similarity

torch.nn.functional.cosine_similarity 是 PyTorch 中的一个函数，用于计算两个张量之间的余弦相似度。这个函数在机器学习和深度学习领域中非常有用，尤其是在处理文本、图像或任何类型的特征向量时，用于度量它们之间的相似性。

用途

• 计算两个向量或向量组之间的余弦相似度。
• 广泛应用于自然语言处理、计算机视觉、推荐系统等领域。

用法

torch.nn.functional.cosine_similarity(x1, x2, dim=1, eps=1e-8)

• x1, x2: 输入的两个张量，必须能够广播到相同的形状。
• dim: 计算相似度的维度。
• eps: 避免除零错误的小数值。

参数

• x1 (Tensor): 第一个输入张量。
• x2 (Tensor): 第二个输入张量。
• dim (int, 可选): 计算相似度的维度，默认为1。
• eps (float, 可选): 用于避免除零的小数值，默认为1e-8。

数学理论

余弦相似度的计算公式为：

 

• x1⋅x2 表示两个张量的点积。
•  和 ​ 分别是 x1 和 x2 的2范数。
• ε 是一个小的数值，用来保证除数不为零。

示例代码 

import torch
import torch.nn.functional as F

# 随机生成两个张量
input1 = torch.randn(100, 128)
input2 = torch.randn(100, 128)

# 计算余弦相似度
output = F.cosine_similarity(input1, input2)

# 打印结果
print(output)

输出结果

此代码将计算 input1 和 input2 每行之间的余弦相似度，并输出一个长度为100的张量，每个元素对应于两个输入张量相应行的余弦相似度值。由于输入是随机生成的，输出也会随机变化。

pdist

torch.nn.functional.pdist 是 PyTorch 中的一个函数，它用于计算输入张量中每对行向量之间的 p 范数距离。此函数在统计分析、机器学习和数据科学中非常有用，尤其是在涉及距离度量和空间关系的场景中。

用途

• 计算给定张量中每对行向量之间的距离。
• 应用于聚类分析、多维缩放和其他需要距离度量的算法。

用法

torch.nn.functional.pdist(input, p=2)

• input: 输入张量，其形状为 N×M，其中 N 是行数，M 是列数（特征数）。
• p: 用于计算的 p 范数，默认为 2，即欧几里得距离。

参数

• input (Tensor): 形状为 N×M 的输入张量。
• p (float): p 范数的值，用于计算向量对之间的距离。可取值为 0 到 ∞ 之间的任何实数。

数学理论

对于输入张量的每一对行向量  和 ，pdist 计算它们之间的 p 范数距离：​ 其中，​ 和  分别是  和的第 k 个元素。

示例代码

import torch
import torch.nn.functional as F

# 定义输入张量
input_tensor = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]], dtype=torch.float32)

# 计算 p 范数距离
distances = F.pdist(input_tensor, p=2)

# 输出结果 tensor([5.1962, 10.3923, 5.1962]) 这里，输出的是输入张量中每一对行向量之间的欧几里得距离。

print(distances)

总结 

本文解析了 PyTorch 中三个关键的距离函数：pairwise_distance、cosine_similarity 和 pdist。这些函数在深度学习和机器学习中非常重要，用于计算向量之间的距离和相似度，从而支持各种算法如聚类、k-最近邻、特征相似度度量等。每个函数都有其特定的应用场景和数学原理。pairwise_distance 计算两组向量间的成对欧几里得距离，cosine_similarity 计算两个张量间的余弦相似度，而 pdist 则计算一个张量内各行向量间的 p 范数距离。通过这些函数，我们能有效地分析和处理数据，特别是在高维空间中。