cuvis\_ai.utils.torch ===================== .. automodule:: cuvis_ai.utils.torch .. rubric:: Functions .. autosummary:: :toctree: extract_state_dict get_output_shape guess_input_dimensionalty guess_state_dict_format .. rubric:: Classes .. autosummary:: :toctree: :template: custom-class-template.rst InputDimension