Función torch.dist de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.distEs una función en PyTorch utilizada para calcular la distancia entre dos tensores.

Definición de la función

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

Descripción de parámetros:

  • input- Tensor de entrada
  • other- Tensor objetivo
  • p- Tipo de norma, por defecto es 2 (distancia euclidiana)

Ejemplo de uso

Ejemplo

import torch

# Crear dos tensores
x = torch.tensor([1.0, 2.0, 3.0])
y = torch.tensor([4.0, 5.0, 6.0])

# Calcular la distancia euclidiana (p=2)
dist = torch.dist(x, y)
print("Distancia euclidiana:", dist)

# Calcular la distancia de Manhattan (p=1)
dist_l1 = torch.dist(x, y, p=1)
print("Distancia de Manhattan:", dist_l1)

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones