code: def tree_map_only(
      __type_or_types: Type2[T, S],
      func: Fn2[T, S, Any],
      tree: PyTree,
      is_leaf: Optional[Callable[[PyTree], bool]] = None,
  ) -> PyTree:
      ...
  
imports: import functools
  import warnings
  from typing import Any, Callable, Iterable, List, Optional, overload, Tuple, Type, TypeVar, Union
  import torch
  import optree
  from optree import PyTreeSpec
label_desc: Use image node_id 17 for calling a global function with example usage: tree_map_only(__type_or_types, func, tree, is_leaf) without return types