PyTorch torch.dist-Funktion

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

torch.distEs ist eine Funktion in PyTorch, mit der die Distanz zwischen zwei Tensoren berechnet wird.

Funktionsdefinition

torch.dist(input, other, p=2)

Parameterbeschreibung:

  • input- Eingabe-Tensor
  • other- Ziel-Tensor
  • p- Normtyp, Standardwert ist 2 (euklidische Distanz)

Anwendungsbeispiel

Beispiel

import torch

# Zwei Tensoren erstellen
x = torch.tensor([1.0, 2.0, 3.0])
y = torch.tensor([4.0, 5.0, 6.0])

# Euklidische Distanz berechnen (p=2)
dist = torch.dist(x, y)
print("Euklidische Distanz:", dist)

# Manhattan-Distanz berechnen (p=1)
dist_l1 = torch.dist(x, y, p=1)
print("Manhattan-Distanz:", dist_l1)

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

Andere Erweiterungen