yobx.torch
==========

.. toctree::
    :maxdepth: 1
    :caption: main functions

    to_onnx

.. toctree::
    :maxdepth: 1
    :caption: modules

    _model_eval_cases
    export_helper
    export_options
    fake_tensor_helper
    flatten
    flatten_helper
    input_observer
    model_eval_cases
    patch
    tiny_models
    torch_helper
    tracing
    validate
    in_torch/index
    in_transformers/index
    interpreter/index

register_flattening_functions
+++++++++++++++++++++++++++++

.. autofunction:: yobx.torch.register_flattening_functions