torchdrug.transforms
====================

.. currentmodule:: torchdrug.transforms

.. autoclass:: Compose

Graph Transformations
---------------------

NormalizeTarget
^^^^^^^^^^^^^^^
.. autoclass:: NormalizeTarget

RandomBFSOrder
^^^^^^^^^^^^^^
.. autoclass:: RandomBFSOrder

Shuffle
^^^^^^^
.. autoclass:: Shuffle

VirtualNode
^^^^^^^^^^^
.. autoclass:: VirtualNode


Molecule Transformations
------------------------

RemapAtomType
^^^^^^^^^^^^^
.. autoclass:: RemapAtomType

VirtualAtom
^^^^^^^^^^^
.. autoclass:: VirtualAtom


Protein Transformations
-----------------------

TruncateProtein
^^^^^^^^^^^^^^^
.. autoclass:: TruncateProtein

ProteinView
^^^^^^^^^^^
.. autoclass:: ProteinView