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

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

    to_onnx

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

    export_helper
    export_options
    fake_tensor_helper
    flatten
    input_observer
    patch
    tiny_models
    torch_helper
    tracing
    validate
    coverage/index
    in_torch/index
    in_transformers/index
    interpreter/index
    new_tracing/index
    testing/index

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

.. autofunction:: yobx.torch.register_flattening_functions