torchdrug.utils
===============

Visualization
-------------

.. currentmodule:: torchdrug.utils.plot

.. autofunction:: reaction

.. autofunction:: highlight

.. autofunction:: echarts

Auxiliary Torch Functions
-------------------------

.. currentmodule:: torchdrug.utils

.. autofunction:: load_extension

.. autofunction:: cpu

.. autofunction:: cuda

.. autofunction:: detach

.. autofunction:: clone

.. autofunction:: mean

.. autofunction:: cat

.. autofunction:: stack

.. autofunction:: sparse_coo_tensor

Distributed Communication
-------------------------

.. currentmodule:: torchdrug.utils.comm

.. autofunction:: init_process_group

.. autofunction:: get_cpu_count

.. autofunction:: get_group

.. autofunction:: get_rank

.. autofunction:: get_world_size

.. autofunction:: reduce

.. autofunction:: stack

.. autofunction:: cat

File Processing
---------------

.. currentmodule:: torchdrug.utils

.. autofunction:: download

.. autofunction:: extract

.. autofunction:: compute_md5

.. autofunction:: get_line_count

Commandline I/O
---------------

.. autofunction:: input_choice

.. autofunction:: literal_eval

.. autofunction:: no_rdkit_log

.. autofunction:: capture_rdkit_log

.. automodule:: torchdrug.utils.pretty
    :members:

Decorator
---------

.. currentmodule:: torchdrug.utils

.. autofunction:: cached_property

Helper functions
----------------

.. currentmodule:: torchdrug.utils

.. autofunction:: copy_args