super(MyModule, self).init()
      def forward(self, input: Any): # note the Any type
          if torch.jit.isinstance(input, List[torch.Tensor]):
              for t in input:
                  y = t.clamp(0, 0.5)
          elif torch.jit.isinstance(input, Dict[str, str]):
              for val in input.values():
                  print(val)

  m = torch.jit.script(MyModule())
  x = [torch.rand(3,3), torch.rand(4,3)]
  m(x)
  y = {"key1":"val1","key2":"val2"}
  m(y)
