PyTorch torch.dist 関数
Pytorch torch リファレンスマニュアルtorch.distは PyTorch で2つのテンソル間の距離を計算するために使用される関数です。
関数定義
torch.dist(input, other, p=2)
パラメータの説明：
input- 入力テンソルother- ターゲットテンソルp- ノルムの種類、デフォルトは2（ユークリッド距離）
使用例
インスタンス
import torch
# 2つのテンソルを作成
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)
# 2つのテンソルを作成
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)
その他の拡張