torch.diag\_embed ================= .. currentmodule:: torch .. autofunction:: diag_embed