docstr: Pass a two dimensional input through a Dropout layer
  that keeps half the input and test
  that half of the output values are zero.
docstr_tok: ["Pass","a","two","dimensional","input","through","a","Dropout","layer","that","keeps","half","the","input","and","test","that","half","of","the","output","values","are","zero","."]
code: def test_on_dropout_two_dimension(self):
      """
      Pass a two dimensional input through a Dropout layer
      that keeps half the input and test
      that half of the output values are zero.
      """
  
      drop_input = jn.array([[1.0, 2.0, 3.0], [5.0, 6.0, 7.0]])
      keep = 0.5
      test_generator = random.DEFAULT_GENERATOR
      test_generator.seed(3)
      dropout_layer = objax.nn.Dropout(keep, test_generator)
      training = True
      drop_output = dropout_layer(drop_input, training)
      self.assertEqual(jn.count_nonzero(drop_output), 3)
  
file_path: objax/tests/dropout.py
imports: import unittest
  import jax.numpy
  import objax
  from objax import random
label_desc: Use image node_id 2 for calling the TestDropout obj's underlying member method code with example usage: obj.test_on_dropout_two_dimension() without return types