[ 仅参数名不一致 ]torch.diagonal

torch.diagonal

torch.diagonal(input,
               offset=0,
               dim1=0,
               dim2=1))

paddle.diagonal

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

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

参数映射

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | input | x | 表示输入的 Tensor ,仅参数名不一致。 | | offset | offset | 表示对角线偏移量。 | | dim1 | axis1 | 获取对角线的二维平面的第一维,仅参数名不一致。 | | dim2 | axis2 | 获取对角线的二维平面的第二维,仅参数名不一致。 |