torchoutil package

Collection of functions and modules to help development in PyTorch.

Subpackages

• torchoutil.core package
  • Submodules
    • torchoutil.core.dtype_enum module
      • DTypeEnum
      • enum_dtype_to_str()
      • enum_dtype_to_torch_dtype()
      • str_to_enum_dtype()
      • str_to_torch_dtype()
      • torch_dtype_to_enum_dtype()
      • torch_dtype_to_str()
    • torchoutil.core.get module
    • torchoutil.core.make module
      • as_device()
      • as_dtype()
      • as_generator()
      • get_default_device()
      • get_default_dtype()
      • get_default_generator()
      • set_default_dtype()
      • set_default_generator()
    • torchoutil.core.packaging module
      • requires_packages()
      • torch_version_ge_1_13()
    • torchoutil.core.semver module
    • torchoutil.core.version module
      • get_githash_full()
      • get_githash_short()
• torchoutil.extras package
  • Subpackages
    • torchoutil.extras.hdf package
      • Submodules
    • torchoutil.extras.numpy package
      • Submodules
  • Submodules
    • torchoutil.extras.colorlog module
      • get_colored_formatter()
    • torchoutil.extras.safetensors module
      • dump_safetensors()
      • load_safetensors()
      • to_safetensors()
    • torchoutil.extras.tensorboard module
      • TensorboardEvent
      • get_duration()
      • get_tfevents_duration()
      • load_event_file()
      • load_event_files()
      • load_tfevents()
      • load_tfevents_files()
      • load_with_tensorboard()
    • torchoutil.extras.torchaudio module
      • dump_audio()
      • dump_with_torchaudio()
      • load_audio()
      • load_with_torchaudio()
      • to_torchaudio()
    • torchoutil.extras.yaml module
      • IgnoreTagLoader
      • SplitTagLoader
      • dump_yaml()
      • load_yaml()
      • to_yaml()
• torchoutil.hub package
  • Submodules
    • torchoutil.hub.download module
      • download_file()
    • torchoutil.hub.paths module
      • get_cache_dir()
      • get_tmp_dir()
    • torchoutil.hub.registry module
      • RegistryEntry
      • RegistryHub
      • get_default_register_root()
• torchoutil.nn package
  • Subpackages
    • torchoutil.nn.functional package
      • Submodules
    • torchoutil.nn.modules package
      • Submodules
• torchoutil.optim package
  • Submodules
    • torchoutil.optim.schedulers module
      • CosDecayRule
      • CosDecayScheduler
    • torchoutil.optim.utils module
      • create_params_groups_bias()
      • get_lr()
      • get_lrs()
• torchoutil.pyoutil package
  • Subpackages
    • torchoutil.pyoutil.typing package
      • Submodules
  • Submodules
    • torchoutil.pyoutil.abc module
      • Singleton
    • torchoutil.pyoutil.argparse module
      • str_to_bool()
      • str_to_optional_bool()
      • str_to_optional_float()
      • str_to_optional_int()
      • str_to_optional_str()
    • torchoutil.pyoutil.collections module
      • all_eq()
      • all_ne()
      • argmax()
      • argmin()
      • contained()
      • dict_list_to_list_dict()
      • dump_dict()
      • filter_iterable()
      • find()
      • flat_dict_of_dict()
      • flat_list_of_list()
      • flatten()
      • intersect()
      • intersect_lists()
      • is_full()
      • is_sorted()
      • is_unique()
      • list_dict_to_dict_list()
      • prod()
      • recursive_generator()
      • reduce_add()
      • reduce_and()
      • reduce_mul()
      • reduce_or()
      • shuffled()
      • sorted_dict()
      • sum()
      • unflat_dict_of_dict()
      • unflat_list_of_list()
      • union()
      • union_dicts()
      • union_lists()
      • unzip()
    • torchoutil.pyoutil.csv module
      • dump_csv()
      • load_csv()
    • torchoutil.pyoutil.dataclasses module
      • get_defaults_values()
    • torchoutil.pyoutil.datetime module
      • now()
      • now_iso()
      • now_iso8601()
    • torchoutil.pyoutil.difflib module
      • find_closest_in_list()
      • sequence_matcher_ratio()
    • torchoutil.pyoutil.enum module
      • StrEnum
    • torchoutil.pyoutil.functools module
      • Compose
      • compose
      • filter_and_call()
      • function_alias()
      • identity()
    • torchoutil.pyoutil.hashlib module
      • Hasher
      • hash_file()
    • torchoutil.pyoutil.importlib module
      • Placeholder
      • is_available_package()
      • is_editable_package()
      • package_is_available()
      • reload_editable_packages()
      • reload_submodules()
      • search_submodules()
    • torchoutil.pyoutil.inspect module
      • get_argnames()
      • get_current_fn_name()
      • get_fullname()
    • torchoutil.pyoutil.io module
      • open_close_wrap()
    • torchoutil.pyoutil.json module
      • dump_json()
      • load_json()
    • torchoutil.pyoutil.logging module
      • MkdirFileHandler
      • get_current_file_logger()
      • get_ipython_name()
      • get_null_logger()
      • log_once()
      • running_on_interpreter()
      • running_on_notebook()
      • running_on_terminal()
      • setup_logging_level()
      • setup_logging_verbose()
      • warn_once()
    • torchoutil.pyoutil.math module
      • clamp()
      • clip()
      • nextafter()
      • nextdown()
    • torchoutil.pyoutil.os module
      • get_num_cpus_available()
      • safe_rmdir()
      • tree_iter()
    • torchoutil.pyoutil.pickle module
      • dump_pickle()
      • load_pickle()
    • torchoutil.pyoutil.random module
      • randstr()
    • torchoutil.pyoutil.re module
      • compile_patterns()
      • find_patterns()
      • get_key_fn()
      • match_patterns()
      • sort_with_patterns()
    • torchoutil.pyoutil.semver module
      • Version
      • VersionDict
    • torchoutil.pyoutil.warnings module
      • deprecated_alias()
      • deprecated_function()
      • warn_once()
• torchoutil.serialization package
  • Submodules
    • torchoutil.serialization.common module
      • to_builtin()
    • torchoutil.serialization.csv module
      • dump_csv()
      • load_csv()
      • to_csv()
    • torchoutil.serialization.dump_fn module
      • dump()
      • save()
    • torchoutil.serialization.hdf module
    • torchoutil.serialization.json module
      • dump_json()
      • to_json()
    • torchoutil.serialization.load_fn module
      • load()
    • torchoutil.serialization.numpy module
    • torchoutil.serialization.pickle module
      • dump_pickle()
      • to_pickle()
    • torchoutil.serialization.safetensors module
    • torchoutil.serialization.torch module
      • dump_torch()
      • load_torch()
      • to_torch()
    • torchoutil.serialization.torchaudio module
    • torchoutil.serialization.yaml module
• torchoutil.types package
  • Submodules
    • torchoutil.types.guards module
      • is_bool_tensor()
      • is_bool_tensor1d()
      • is_complex_tensor()
      • is_dict_str_tensor()
      • is_floating_tensor()
      • is_integral_dtype()
      • is_integral_tensor()
      • is_integral_tensor1d()
      • is_iterable_tensor()
      • is_list_tensor()
      • is_number_like()
      • is_scalar_like()
      • is_tensor0d()
      • is_tensor_like()
      • is_tensor_or_array()
      • is_tuple_tensor()
    • torchoutil.types.tensor_subclasses module
      • BoolTensor
      • BoolTensor0D
      • BoolTensor1D
      • BoolTensor2D
      • BoolTensor3D
      • ByteTensor
      • ByteTensor0D
      • ByteTensor1D
      • ByteTensor2D
      • ByteTensor3D
      • CDoubleTensor
      • CDoubleTensor0D
      • CDoubleTensor1D
      • CDoubleTensor2D
      • CDoubleTensor3D
      • CFloatTensor
      • CFloatTensor0D
      • CFloatTensor1D
      • CFloatTensor2D
      • CFloatTensor3D
      • CHalfTensor
      • CHalfTensor0D
      • CHalfTensor1D
      • CHalfTensor2D
      • CHalfTensor3D
      • CharTensor
      • CharTensor0D
      • CharTensor1D
      • CharTensor2D
      • CharTensor3D
      • ComplexFloatingTensor
      • ComplexFloatingTensor0D
      • ComplexFloatingTensor1D
      • ComplexFloatingTensor2D
      • ComplexFloatingTensor3D
      • DoubleTensor
      • DoubleTensor0D
      • DoubleTensor1D
      • DoubleTensor2D
      • DoubleTensor3D
      • FloatTensor
      • FloatTensor0D
      • FloatTensor1D
      • FloatTensor2D
      • FloatTensor3D
      • FloatingTensor
      • FloatingTensor0D
      • FloatingTensor1D
      • FloatingTensor2D
      • FloatingTensor3D
      • HalfTensor
      • HalfTensor0D
      • HalfTensor1D
      • HalfTensor2D
      • HalfTensor3D
      • IntTensor
      • IntTensor0D
      • IntTensor1D
      • IntTensor2D
      • IntTensor3D
      • IntegralTensor
      • IntegralTensor0D
      • IntegralTensor1D
      • IntegralTensor2D
      • IntegralTensor3D
      • LongTensor
      • LongTensor0D
      • LongTensor1D
      • LongTensor2D
      • LongTensor3D
      • ShortTensor
      • ShortTensor0D
      • ShortTensor1D
      • ShortTensor2D
      • ShortTensor3D
      • SignedIntegerTensor
      • SignedIntegerTensor0D
      • SignedIntegerTensor1D
      • SignedIntegerTensor2D
      • SignedIntegerTensor3D
      • Tensor
      • Tensor0D
      • Tensor1D
      • Tensor2D
      • Tensor3D
      • UnsignedIntegerTensor
      • UnsignedIntegerTensor0D
      • UnsignedIntegerTensor1D
      • UnsignedIntegerTensor2D
      • UnsignedIntegerTensor3D
    • torchoutil.types.variable_fns module
• torchoutil.utils package
  • Subpackages
    • torchoutil.utils.data package
      • Submodules
  • Submodules
    • torchoutil.utils.hdf module
    • torchoutil.utils.return_types module
      • max
      • min
      • ndim
      • shape
      • sort
      • top_p
      • topk
    • torchoutil.utils.saving module

Submodules

• torchoutil.entries module
  • main_safe_rmdir()
  • main_tree()
  • print_safe_rmdir()
  • print_tree()
• torchoutil.info module
  • get_install_info()
  • get_package_repository_path()
  • print_install_info()