torch.outer
- torch.outer(input, vec2, *, out=None) → Tensor
-
input
和vec2
的外积。如果input
是大小为 $n$ 的向量,vec2
是大小为 $m$ 的向量,则输出矩阵out
必须是大小为 $(n \times m)$。注意
此函数不会进行广播。
示例:
>>> v1 = torch.arange(1., 5.) >>> v2 = torch.arange(1., 4.) >>> torch.outer(v1, v2) tensor([[ 1., 2., 3.], [ 2., 4., 6.], [ 3., 6., 9.], [ 4., 8., 12.]])