torch.Tensor.diag\_embed ======================== .. currentmodule:: torch .. automethod:: Tensor.diag_embed