docstr: Tests that the forward pass runs and returns the expected shape.
docstr_tok: ["Tests","that","the","forward","pass","runs","and","returns","the","expected","shape","."]
code: def test_residual_layer_forward(self):
      """Tests that the forward pass runs and returns the expected shape."""
      model = rezero.ResidualZero(tl.Dense(5))
      x = [np.arange(5).astype(np.float32)]
      _, _ = model.init(shapes.signature(x))
      y = model(x)
      self.assertEqual(y.tolist(), [0.0, 1.0, 2.0, 3.0, 4.0])
  
file_path: trax/trax/models/research/rezero_test.py
imports: from absl.testing import absltest
  import numpy
  from trax import layers
  from trax import shapes
  from trax.models.research import rezero
label_desc: Use image node_id 1 for calling the ResidualZeroTest obj's underlying member method code with example usage: obj.test_residual_layer_forward() without return types