[torch 参数更多]torch.linalg.vecdot¶

torch.linalg.vecdot ¶

          torch.linalg.vecdot(x, y, *, dim=-1, out=None)

         

          paddle.linalg.vecdot(x, y, axis=-1, name=None)

         

其中 PyTorch 相比 Paddle 支持更多其他参数，具体如下：

PyTorch	PaddlePaddle	备注
x	x	输入 Tensor。
y	y	输入 Tensor。
dim	axis	计算向量点积的维度，仅参数名不一致。
out	-	表示输出的 Tensor ， Paddle 无此参数，需要转写。

           # PyTorch 写法
torch.linalg.vecdot(x=a, y=b, out=out)

# Paddle 写法
paddle.assign(paddle.linalg.vecdot(x=a, y=b), output=out)