code: def testVarianceScalingMultiAxis(self):
      rng = random.PRNGKey(0)
      shape = (2, 3, 4, 5)
      initializer = nn.initializers.variance_scaling(
          scale=1.0,
          mode="fan_avg",
          distribution="truncated_normal",
          in_axis=(0, 1),
          out_axis=(-2, -1),
      )
      val = initializer(rng, shape)
  
      self.assertEqual(shape, jnp.shape(val))
  
imports: import collections
  from functools import partial
  import itertools
  import unittest
  from absl.testing import absltest
  from absl.testing import parameterized
  import scipy.stats
  from jax._src import config
  from jax._src import core
  from jax._src import test_util
  from jax._src import ad_checkpoint
  from jax.test_util import check_grads
  from jax import nn
  from jax import random
  import jax
  import jax.numpy
label_desc: Use image node_id 3 for calling the NNInitializersTest obj's underlying member method code with example usage: obj.testVarianceScalingMultiAxis() without return types