tf.keras.tree.map_shape_structure

Variant of keras.tree.map_structure that operates on shape tuples.

Tuples containing ints and Nones are considered shapes and passed to func.

structure Arbitrarily nested structure.

The same structure with func applied.