PyTorch torch.vdot 関数
torch.vdotは、PyTorch で2つのベクトルのドット積(ベクトル内積とも呼ばれる)を計算する関数です。とtorch.dot異なり、torch.vdot複素数ベクトルを処理でき、最初のベクトルの各要素と2番目のベクトルの共役要素を自動的に掛け算します。
関数定義
torch.vdot(input, other)
使用例
例
import torch
# 実数ベクトル
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])
result = torch.vdot(a, b)
print(vdot 結果:, result)
# 1*4 + 2*5 + 3*6 = 32
# 複素数ベクトル
c = torch.tensor([1+2j, 3+4j])
d = torch.tensor([5+6j, 7+8j])
result_complex = torch.vdot(c, d)
print(複素数 vdot 結果:, result_complex)
# 実数ベクトル
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])
result = torch.vdot(a, b)
print(vdot 結果:, result)
# 1*4 + 2*5 + 3*6 = 32
# 複素数ベクトル
c = torch.tensor([1+2j, 3+4j])
d = torch.tensor([5+6j, 7+8j])
result_complex = torch.vdot(c, d)
print(複素数 vdot 結果:, result_complex)
出力結果は次のとおりです:
vdot 结果: tensor(32) 复数 vdot 结果: tensor(70.-6.j)
その他の拡張
Pytorch torch リファレンスマニュアル