code: def test_normalized_rss_initializer():
      target_rss = 0.5
      units = 10
      dense = mx.gluon.nn.Dense(
          units=units,
          weight_initializer=NormalizedRSSInitializer(target_rss),
      )
      dense.initialize()
  
      input_data = mx.random.uniform(shape=(25, 5))
      output_data = dense(input_data)
  
      weights = dense.weight.data()
      assert weights.shape == (10, 5)
      rss = weights.square().sum(axis=1).sqrt()
      np.testing.assert_almost_equal(
          rss.asnumpy(), np.tile(target_rss, units)
      )
  
imports: import mxnet
  import numpy
  import os
  import pytest
  import sys
  from rl_coach.architectures.mxnet_components.heads.head import NormalizedRSSInitializer
label_desc: Use image node_id 1 for calling a global function with example usage: test_normalized_rss_initializer() without return types