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

torch.Tensor.diagonal

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

paddle.Tensor.diagonal

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

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

参数映射

PyTorch PaddlePaddle 备注
offset offset 从指定的二维平面中获取对角线的位置,默认值为 0,既主对角线,仅参数名不一致。
dim1 axis1 获取对角线的二维平面的第一维,默认值为 0,仅参数名不一致。
dim2 axis2 获取对角线的二维平面的第二维,默认值为 1,仅参数名不一致。