PyTorch torch.vdot 関数


Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

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)

出力結果は次のとおりです:

vdot 结果: tensor(32)
复数 vdot 结果: tensor(70.-6.j)

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

その他の拡張