PyTorch torch.ger 関数


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

torch.gerPyTorch で2つのベクトルの外積(グラム行列)を計算するために使用される関数です。

関数定義

torch.ger(input, vec2, out=None)

使用例

実例

import torch

# 2つのベクトルを作成
vec1 = torch.tensor([1, 2, 3])
vec2 = torch.tensor([4, 5, 6])

# 外積を計算
result = torch.ger(vec1, vec2)
print("外積の結果:")
print(result)

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

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

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

その他の拡張