Function get_shape_hash

imperative/python/megengine/jit/partial_tracing.py:175–179

(*tensors)

                    return rst

        def get_shape_hash(*tensors):
            def map_scalar_to_tuple(ishape):
                return (1,) if ishape == tuple() else ishape

            return hash(tuple([map_scalar_to_tuple(t._tuple_shape) for t in tensors]))

        def wrapped_func(*args, **kwargs):
            from ..traced_module.pytree import tree_flatten