def METHOD_NAME(self, M, example_inputs, expected_fqn_to_dim):
    m = M().eval()
    fqn_to_example_inputs = get_fqn_to_example_inputs(m, example_inputs)
    for fqn, expected_dims in expected_fqn_to_dim.items():
        assert fqn in expected_fqn_to_dim
        example_inputs = fqn_to_example_inputs[fqn]
        for example_input, expected_dim in zip(example_inputs, expected_dims):
            assert example_input.dim() == expected_dim