torch.dist
函数是 PyTorch 库中用来计算两个张量之间的距离的函数。- 该函数主要用来计算指定阶数的距离,比如欧几里得距离(默认情况下)
torch.dist(input, other, p=2)
input | 第一个输入张量 |
other | 第二个输入张量,需要与第一个张量的大小相同 |
p | 计算距离时使用的范数的阶数。 默认值是 当
|
import torch
# Define the points
point1 = torch.tensor([1.0, 2.0, 3.0])
point2 = torch.tensor([4.0, 6.0, 8.0])
distance = torch.dist(point1, point2)
print(distance)