[ 参数完全一致 ]torch.Tensor.diag_embed

torch.Tensor.diag_embed

torch.Tensor.diag_embed(offset=0, dim1=-2, dim2=-1)

paddle.Tensor.diag_embed

paddle.Tensor.diag_embed(offset=0, dim1=-2, dim2=-1, name=None)

功能一致,参数完全一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
offset offset 从指定的二维平面中获取对角线的位置。
dim1 dim1 填充对角线的二维平面的第一维。
dim2 dim2 填充对角线的二维平面的第二维。