[ torch 参数更多]torch.logcumsumexp¶

torch.logcumsumexp ¶

          torch.logcumsumexp(input, dim, *, out=None)

         

          paddle.logcumsumexp(x, axis=None, dtype=None, name=None)

         

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

PyTorch	PaddlePaddle	备注
input	x	表示输入的 Tensor ，仅参数名不一致。
dim	axis	表示需要计算的维度，仅参数名不一致。
out	-	表示输出的 Tensor，Paddle 无此参数，需要转写。
-	dtype	表示输出 Tensor 的数据类型，Pytorch 无此参数，Paddle 保持默认即可。

           # Pytorch 写法
torch.logcumsumexp(x,dim=0, out=output)

# Paddle 写法
paddle.assign(paddle.logcumsumexp(x,axis=0), output)