torch.Tensor.diag_embed

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

参见 torch.diag_embed()