code: def tensor_shapes_to_shape_dtypes(shapes, dtype):
      return math_lib.nested_map(
          lambda s: shapes_lib.ShapeDtype(s.as_list(), dtype), shapes
      )
  
file_path: trax/trax/trax2keras.py
imports: import functools
  import tensorflow.compat.v2
  from trax import fastmath
  from trax import shapes
  from trax.fastmath import numpy
  from trax.layers import base
label_desc: Use image node_id 2 for calling a global function with example usage: tensor_shapes_to_shape_dtypes(shapes, dtype) and returns: math_lib