PyTorch torch.outer 関数


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

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)

出力結果は:

tensor([[ 4,  5,  6],
        [ 8, 10, 12],
        [12, 15, 18]])

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

その他の拡張