PyTorch torch.outer 関数
torch.outerPyTorch において、2つのベクトルの外積(クロス積またはテンソル積とも呼ばれる)を計算するための関数です。外積は、一方のベクトルの各要素と他方のベクトルのすべての要素を掛け合わせて、行列を生成します。
関数定義
torch.outer(vec1, vec2, out)
使用例
実例
import torch
# 2つのベクトルを定義
vec1 = torch.tensor([1, 2, 3])
vec2 = torch.tensor([4, 5, 6])
# 外積を計算
result = torch.outer(vec1, vec2)
print(result)
# 2つのベクトルを定義
vec1 = torch.tensor([1, 2, 3])
vec2 = torch.tensor([4, 5, 6])
# 外積を計算
result = torch.outer(vec1, vec2)
print(result)
出力結果は:
tensor([[ 4, 5, 6],
[ 8, 10, 12],
[12, 15, 18]])
その他の拡張
Pytorch torch リファレンスマニュアル