code: def test_resnet(self):
      model = resnet.Resnet50(d_hidden=8, n_output_classes=10)
      x = np.ones((3, 256, 256, 3)).astype(np.float32)
      _, _ = model.init(shapes.signature(x))
      y = model(x)
      self.assertEqual(y.shape, (3, 10))
  
file_path: trax/trax/models/resnet_test.py
imports: from absl.testing import absltest
  import numpy
  from trax import fastmath
  from trax import shapes
  from trax.models import resnet
label_desc: Use image node_id 1 for calling the ResnetTest obj's underlying member method code with example usage: obj.test_resnet() without return types