code: def foo():
      # W = torch.rand(size=(2000, 2000))
      W = torch.rand(size=(2000, 2000, 10, 8))
      b = torch.rand(size=(2000, 1))
      h = torch.rand(size=(1_000_000,))
      x = torch.rand(size=(2000, 1))
      # g = tsensor.astviz("b = W@b + (h+3).dot(h) + torch.abs(torch.tensor(34))",
      #                    sys._getframe())
      frame = sys._getframe()
      frame = None
      g = tsensor.astviz(
          "b = W[:,:,0,0]@b + (h+3).dot(h) + torch.abs(torch.tensor(34))",
          frame,
      )
      g.view()
  
file_path: tensor-sensor/testing/viz_testing.py
imports: import sys
  import torch
  import numpy
  import graphviz
  import tempfile
  import matplotlib.patches
  import matplotlib.pyplot
  import matplotlib.font_manager
  import tsensor
  import torch
  import tsensor
label_desc: Use image node_id 1 for calling a global function with example usage: foo() without return types