torch.outer¶
- torch.outer(input, vec2, *, out=None) Tensor ¶
input
和vec2
的外积。如果input
是大小为 的向量,vec2
是大小为 的向量,那么out
必须是一个大小为 的矩阵。注意
此功能不广播。
- 参数:
输入(张量)- 1-D 输入向量
vec2(张量)- 1-D 输入向量
- 关键字参数:
out(张量,可选)- 可选输出矩阵
示例:
>>> 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.]])