[ 组合替代实现 ]torch.addbmm¶

torch.addbmm ¶

          torch.addbmm(input, batch1, batch2, *, beta=1, alpha=1, out=None)

         

Paddle 无此 API，需要组合实现。

转写示例¶

          # Pytorch 写法
y = torch.addbmm(input, batch1, batch2, beta=beta, alpha=alpha)

# Paddle 写法
y = beta * input + alpha * paddle.sum(paddle.bmm(batch1, batch2), axis=0)