code: def test_message_norm():
      norm = MessageNorm(learn_scale=True)
      assert str(norm) == "MessageNorm(learn_scale=True)"
      x = torch.randn(100, 16)
      msg = torch.randn(100, 16)
      out = norm(x, msg)
      assert out.size() == (100, 16)
  
      if is_full_test():
          jit = torch.jit.script(norm)
          assert torch.allclose(jit(x, msg), out)
  
      norm = MessageNorm(learn_scale=False)
      assert str(norm) == "MessageNorm(learn_scale=False)"
      out = norm(x, msg)
      assert out.size() == (100, 16)
  
      if is_full_test():
          jit = torch.jit.script(norm)
          assert torch.allclose(jit(x, msg), out)
  
label_desc: Use image node_id 1 for calling a global function with example usage: test_message_norm() without return types