code: def test_prelu(self):
      rng = random.key(0)
      key, skey_1, skey_2 = jax.random.split(rng, 3)
      x = jax.random.uniform(skey_1, (4, 6, 5)) - 0.5
      act = nn.PReLU()
      y, params = act.init_with_output(skey_2, x)
      expected_y = jnp.where(x < 0, x * act.negative_slope_init, x)
      init_negative_slope = params["params"]["negative_slope"]
      expected_negative_slope = jnp.array(
          act.negative_slope_init, dtype=jnp.float32
      )
  
      self.assertEqual(y.shape, x.shape)
      np.testing.assert_array_almost_equal(expected_y, y)
      np.testing.assert_array_equal(
          init_negative_slope, expected_negative_slope
      )
  
file_path: flax/tests/linen/linen_activation_test.py
imports: import jax
  import jax.numpy
  import numpy
  from absl.testing import absltest
  from jax import random
  from flax import linen
label_desc: Use image node_id 1 for calling the ActivationTest obj's underlying member method code with example usage: obj.test_prelu() without return types