Fonction torch.dist de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.distC'est une fonction de PyTorch utilisée pour calculer la distance entre deux tenseurs.

Définition de fonction

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

Description des paramètres :

  • input- Tenseur d'entrée
  • other- Tenseur cible
  • p- Type de norme, par défaut 2 (distance euclidienne)

Exemple d'utilisation

Exemple

import torch

# Créer deux tenseurs
x = torch.tensor([1.0, 2.0, 3.0])
y = torch.tensor([4.0, 5.0, 6.0])

# Calculer la distance euclidienne (p=2)
dist = torch.dist(x, y)
print("Distance euclidienne :", dist)

# Calculer la distance de Manhattan (p=1)
dist_l1 = torch.dist(x, y, p=1)
print("Distance de Manhattan :", dist_l1)

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions