code: def test_fast_eval_shape_dropout(self):
      f = lambda rng, x: basic.dropout(rng, 0.5, x)
      rng = jax.random.PRNGKey(0)
      x = jnp.ones([1])
      y_slow = jax.eval_shape(f, rng, x)
      y_fast = eval_shape.fast_eval_shape(f, rng, x)
      self.assertEqual(y_slow, y_fast)
  
file_path: dm-haiku/haiku/_src/eval_shape_test.py
imports: from absl.testing import absltest
  from absl.testing import parameterized
  from haiku._src import basic
  from haiku._src import eval_shape
  from haiku._src import stateful
  from haiku._src import test_utils
  from haiku._src import transform
  import jax
  import jax.numpy
label_desc: Use image node_id 1 for calling the EvalShapeTest obj's underlying member method code with example usage: obj.test_fast_eval_shape_dropout() without return types