code: def map_only(
      __type_or_types: Type2[T, S]
  ) -> MapOnlyFn[Fn2[T, S, Any]]:
      ...
  
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 11 for calling a global function with example usage: map_only(__type_or_types) without return types