[torch 参数更多]torch.Tensor.norm
torch.Tensor.norm
torch.Tensor.norm(p='fro', dim=None, keepdim=False, dtype=None)
paddle.Tensor.norm
paddle.Tensor.norm(p='fro', axis=None, keepdim=False, name=None)
PyTorch 相比 Paddle 支持更多其他参数,具体如下:
参数映射
PyTorch | PaddlePaddle | 备注 |
---|---|---|
p | p | 范数(ord)的种类。 |
dim | axis | 使用范数计算的轴,仅参数名不一致。 |
keepdim | keepdim | 是否在输出的 Tensor 中保留和输入一样的维度。 |
dtype | - | 输出数据类型,Paddle 无此参数,需要转写。 |
转写示例
dtype 参数:输出数据类型
# PyTorch 写法
x.norm(dim=-1, dtype=torch.float32)
# Paddle 写法
y = x.astype('float32')
y.norm(dim=-1)