class_docstr: Tests for utilities in nn_ops.py.
class_docstr_tok: ["Tests","for","utilities","in","nn_ops.py","."]
docstr: Tests upscale2x_nearest_neighbor.
docstr_tok: ["Tests","upscale2x_nearest_neighbor","."]
code: def test_upscale2x_nearest_neighbor(self):
      """Tests upscale2x_nearest_neighbor."""
      inputs = jnp.array(np.random.normal(size=(16, 32, 32, 128)))
  
      outputs = nn_ops.upscale2x_nearest_neighbor(inputs)
      # check the output shape
      self.assertEqual(outputs.shape, (16, 64, 64, 128))
  
imports: import functools
  from absl.testing import absltest
  from absl.testing import parameterized
  import flax.linen
  import jax
  import jax.numpy
  import numpy
  from scenic.model_lib.layers import nn_ops
label_desc: Use image node_id 5 for calling the NNOpsTest obj's underlying member method code with example usage: obj.test_upscale2x_nearest_neighbor() without return types