[ 参数完全一致 ]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 | 填充对角线的二维平面的第二维。 |