PyTorch torch.outer 函数
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)
# 定义两个向量
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 参考手册