code: def make_mlp_and_input():
      model = nn.Sequential()
      model.add_module("W0", nn.Linear(8, 16))
      model.add_module("tanh", nn.Tanh())
      model.add_module("W1", nn.Linear(16, 1))
      x = torch.randn(1, 8)
      return model, x
  
file_path: pytorchviz/test/test.py
imports: import unittest
  import torch
  from torch import nn
  from torchviz import make_dot, make_dot_from_trace
label_desc: Use image node_id 1 for calling a global function with example usage: make_mlp_and_input() and returns: model,  x