def test_incompatible_higher_rank_inputs_no_raise(self):
    """The globals can have an arbitrary shape in the input."""
    input_graph = self._get_shaped_input_graph()
    edge_model_fn, node_model_fn, _ = self._get_shaped_model_fns()
    input_graph = input_graph.replace(
        globals=tf.transpose(input_graph.globals, [0, 2, 1, 3]))
    graph_network = modules.InteractionNetwork(edge_model_fn, node_model_fn)
    self._assert_build_and_run(graph_network, input_graph)
The globals can have an arbitrary shape in the input.
graph_nets/tests/modules_test.py