现在位置: 首页 > PyTorch 教程 > 正文

PyTorch torch.ger 函数


Pytorch torch 参考手册 Pytorch torch 参考手册

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)

输出结果为:

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

Pytorch torch 参考手册 Pytorch torch 参考手册