PyTorch torch.ger 函数
torch.ger 是 PyTorch 中用于计算两个向量的外积(Gram 矩阵)的函数。
函数定义
torch.ger(input, vec2, out=None)
使用示例
实例
import torch
# 创建两个向量
vec1 = torch.tensor([1, 2, 3])
vec2 = torch.tensor([4, 5, 6])
# 计算外积
result = torch.ger(vec1, vec2)
print("外积结果:")
print(result)
# 创建两个向量
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 参考手册