openfold.utils.tensor\_utils ============================ .. automodule:: openfold.utils.tensor_utils .. rubric:: Functions .. autosummary:: add batched_gather dict_map dict_multimap flatten_final_dims masked_mean maybe_to one_hot permute_final_dims pts_to_distogram tree_map