deepfold.utils.tensor\_utils
============================

.. automodule:: deepfold.utils.tensor_utils

   
   .. rubric:: Functions

   .. autosummary::
   
      add
      batched_gather
      collate
      dict_map
      dict_multimap
      flatten_final_dims
      masked_mean
      one_hot
      permute_final_dims
      pts_to_distogram
      tree_map