code: def test_graph_size_norm():
      x = torch.randn(100, 16)
      batch = torch.repeat_interleave(
          torch.full((10,), 10, dtype=torch.long)
      )
  
      norm = GraphSizeNorm()
      assert str(norm) == "GraphSizeNorm()"
  
      out = norm(x, batch)
      assert out.size() == (100, 16)
  
      if is_full_test():
          jit = torch.jit.script(norm)
          assert torch.allclose(jit(x, batch), out)
  
label_desc: Use image node_id 1 for calling a global function with example usage: test_graph_size_norm() without return types