torch.Tensor.diag_embed

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

See torch.diag_embed()