code: def test_smoke(batch_size):
      x = torch.rand(batch_size, 2, 3)
      assert x.shape == (batch_size, 2, 3), x.shape
  
label_desc: Use image node_id 1 for calling a global function with example usage: test_smoke(batch_size) without return types