oneflow.diag ============ .. currentmodule:: oneflow .. autofunction:: diag