code: def __init__(self):
      self.W = torch.rand(size=(2, 20, 2000, 10))
      self.b = torch.rand(size=(20, 1))
      # self.x = torch.tensor([4, 5]).reshape(2, 1)
      self.h = torch.rand(size=(1_000_000,))
      self.a = 3
      print(self.W.shape)
      print(self.W[:, :, 1].shape)
  
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 to create a new GRU object  with example: obj = GRU()