torch.block\_diag ================= .. currentmodule:: torch .. autofunction:: block_diag