

name (str, optional) -- name of weight parameter

dim (int, optional) -- dimension over which to
    compute the norm



Returns:
      The original module with the weight norm hook
Return type:
      T_module
Example:
  >>> m = weight_norm(nn.Linear(20, 40), name='weight')
  >>> m
  Linear(in_features=20, out_features=40, bias=True)
  >>> m.weight_g.size()
  torch.Size([40, 1])
  >>> m.weight_v.size()
  torch.Size([40, 20])
