[ 组合替代实现 ]torch.addcmul¶
torch.addcmul¶
torch.addcmul(input, tensor1, tensor2, *, value=1, out=None)
用于实现矩阵 tensor1
与矩阵 tensor2
相乘,再加上输入 input
,公式为:
$ out = input + value * tensor1 * tensor2 $
PaddlePaddle 目前无对应 API,可使用如下代码组合实现该 API。
转写示例¶
# PyTorch 写法
y = torch.addcmul(input, tensor1, tensor2, value=value)
# Paddle 写法
y = input + value * tensor1 * tensor2