code: def test_mlp_forward_shape(self):
      model = mlp.MLP(layer_widths=(32, 16, 8))
      x = np.ones((7, 28, 28, 3)).astype(np.float32)
      _, _ = model.init(shapes.signature(x))
      y = model(x)
      self.assertEqual(y.shape, (7, 8))
  
imports: from absl.testing import absltest
  import numpy
  from trax import fastmath
  from trax import shapes
  from trax.models import mlp
label_desc: Use image node_id 1 for calling the MLPTest obj's underlying member method code with example usage: obj.test_mlp_forward_shape() without return types