code: def test_2d_weight_normalization():
      """Test normalization of weights."""
      w = np.random.rand(10, 5)
      w_out = normalize_weight_shape(w, n_samples=10, n_tasks=5)
      assert w_out.shape == (10, 5)
      assert np.all(w_out == w)
  
imports: import numpy
  import deepchem
  from deepchem.metrics import to_one_hot
  from deepchem.metrics import from_one_hot
  from deepchem.metrics import threshold_predictions
  from deepchem.metrics import handle_classification_mode
  from deepchem.metrics import normalize_prediction_shape
  from deepchem.metrics import normalize_weight_shape
label_desc: Use image node_id 18 for calling a global function with example usage: test_2d_weight_normalization() without return types