[ 仅参数默认值不一致 ]torch.linalg.diagonal

torch.linalg.diagonal

torch.linalg.diagonal(A, *, offset=0, dim1=-2, dim2=-1)

paddle.diagonal

paddle.diagonal(x,
                offset=0,
                axis1=0,
                axis2=1,
                name=None)

两者功能一致且参数用法一致,仅参数默认值不一致,具体如下:

参数映射

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | A | x | 表示输入的 Tensor ,仅参数名不一致。 | | offset | offset | 表示对角线偏移量。 | | dim1 | axis1 | 获取对角线的二维平面的第一维,仅参数默认值不一致。Pytorch 默认为-2,Paddle 默认为0,Paddle 需设置为与 Pytorch 一致。 | | dim2 | axis2 | 获取对角线的二维平面的第二维,仅参数默认值不一致。Pytorch 默认为-1,Paddle 默认为1,Paddle 需设置为与 Pytorch 一致。 |