[ 仅参数名不一致 ]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,仅参数名不一致。 |