addmm
函数将两个张量相乘后相加,其中一个张量是稠密矩阵,另一个张量是稠密向量,并加上一个偏置项。
output = torch.addmm(beta=1, mat, alpha=1, vec1, vec2, out=None)
beta
(float, optional) - 缩放输入稠密矩阵的因子。默认值: 1
。mat
(Tensor) - 乘积的左边张量。维度为 (m, n)。alpha
(float, optional) - 缩放第一个矢量的因子。默认值: 1
。vec1
(Tensor) - 乘积的右边张量。维度为 (n,)。vec2
(Tensor) - 加到乘积结果中的偏置项。维度为 (m,)。out
(Tensor, optional) - 结果张量。output
(Tensor) - 张量输出的形状为 (m,)。import torch
mat = torch.randn(2, 3)
vec1 = torch.randn(3)
vec2 = torch.randn(2)
output = torch.addmm(mat, vec1, vec2)
注意:这里的示例程序以 PyTorch 的语法为准,但本函数为 Open3D 中的张量操作函数,并不完全适用于 PyTorch,只用于对 Open3D 中的函数进行说明。