Функция PyTorch torch.dist

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.distЭто функция в PyTorch, используемая для вычисления расстояния между двумя тензорами.

Определение функции

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

Описание параметров:

  • input- Входной тензор
  • other- Целевой тензор
  • p- Тип нормы, по умолчанию 2 (евклидово расстояние)

Пример использования

Пример

import torch

# Создайте два тензора
x = torch.tensor([1.0, 2.0, 3.0])
y = torch.tensor([4.0, 5.0, 6.0])

# Вычислить евклидово расстояние (p=2)
dist = torch.dist(x, y)
print("Евклидово расстояние:", dist)

# Вычислить манхэттенское расстояние (p=1)
dist_l1 = torch.dist(x, y, p=1)
print("Манхэттенское расстояние:", dist_l1)

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения