PyTorch torch.ger 関数
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)
# 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 リファレンスマニュアル