torch.diag ========== .. currentmodule:: torch .. autofunction:: diag