torch.addr¶
- torch.addr(input, vec1, vec2, *, beta=1, alpha=1, out=None) Tensor ¶
执行向量
vec1
和vec2
的外积,并将其添加到矩阵input
中。可选值
beta
和alpha
分别是vec1
和vec2
之间的外积以及添加的矩阵input
的缩放因子。如果
beta
为 0,则将忽略input
,并且不会传播其中的 nan 和 inf。如果
vec1
是大小为 n 的向量,而vec2
是大小为 m 的向量,则input
必须与大小为 的矩阵广播兼容,并且out
将是大小为 的矩阵。- 参数
- 关键字参数
beta (数字, 可选) –
input
的乘数 ()alpha (数字, 可选) – 的乘数 ()
out (张量, 可选) – 输出张量。
示例
>>> vec1 = torch.arange(1., 4.) >>> vec2 = torch.arange(1., 3.) >>> M = torch.zeros(3, 2) >>> torch.addr(M, vec1, vec2) tensor([[ 1., 2.], [ 2., 4.], [ 3., 6.]])