deepfold.utils.tensor_utils.dict_map

deepfold.utils.tensor_utils.dict_map(fn: Callable[[T], Any], dic: Dict[Any, dict | list | tuple | T], leaf_type: Type[T]) → Dict[Any, dict | list | tuple | Any]