torch.Tensor.diag_embed¶ Tensor.diag_embed(offset=0, dim1=-2, dim2=-1) → Tensor¶ 参见 torch.diag_embed()