[ torch 参数更多 ]torch.Tensor.mean¶

torch.Tensor.mean ¶

          torch.Tensor.mean(dim=None, keepdim=False, *, dtype=None)

         

          paddle.Tensor.mean(axis=None, keepdim=False, name=None)

         

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

PyTorch	PaddlePaddle	备注
dim	axis	指定对 x 进行计算的轴，仅参数名不一致。
keepdim	keepdim	是否在输出 Tensor 中保留减小的维度。
dtype	-	输出 Tensor 的类型，Paddle 无此参数, 需要转写。

           # PyTorch 写法
x.mean(dtype=torch.float32)

# Paddle 写法
x.mean().astype(paddle.float32)