[model_explorer] Allow to pass a config

The idea would be to pass a dictionnary that would parametrize the way the graph is viewed:

• Allow to view attributes by using a lambda: Example:

{
  'attributes': {'dims': lambda tensor: str(tensor.dims())},
  ...