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

PyTorch torch.outer 函数


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

torch.outer 是 PyTorch 中用于计算两个向量的外积(也称为叉积或张量积)的函数。外积会将一个向量中的每个元素与另一个向量中的所有元素相乘,生成一个矩阵。

函数定义

torch.outer(vec1, vec2, out)

使用示例

实例

import torch

# 定义两个向量
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 参考手册