tiny\_diff.utils ================ .. automodule:: tiny_diff.utils .. rubric:: Functions .. autosummary:: :toctree: loss_backward match_shape