code: def test_normalize_weight_linear(self):
      input_dim = 32
      output_dim = 64
      linear = nn.Linear(input_dim, output_dim, bias=False)
      normalize_weight(linear.weight, dim=0)
      self.assertEqual(linear.weight.norm(dim=0).sum(), input_dim)
      normalize_weight(linear.weight, dim=1)
      self.assertEqual(linear.weight.norm(dim=1).sum(), output_dim)
  
imports: import copy
  import unittest
  import torch
  import torch.nn
  from lightly.models import utils
  from lightly.models.utils import _no_grad_trunc_normal, activate_requires_grad, batch_shuffle, batch_unshuffle, deactivate_requires_grad, nearest_neighbors, normalize_weight, update_momentum
label_desc: Use image node_id 5 for calling the TestModelUtils obj's underlying member method code with example usage: obj.test_normalize_weight_linear() without return types