code: def test_set_output_size(
      kernel_sizes, hidden_channels, strides, pool_shape, pool_stride
  ):
      spec = InOutSpec(
          akro.Box(shape=[3, 19, 15], high=np.inf, low=-np.inf),
          akro.Box(shape=[200], high=np.inf, low=-np.inf),
      )
      model = CNNModule(
          spec,
          image_format="NCHW",
          hidden_channels=hidden_channels,
          kernel_sizes=kernel_sizes,
          strides=strides,
          pool_shape=[(pool_shape, pool_shape)],
          pool_stride=[(pool_stride, pool_stride)],
          layer_normalization=True,
      )
      images = torch.ones(10, 3, 19, 15)
      x = model(images)
      assert x.shape == (10, 200)
  
file_path: garage/tests/garage/torch/modules/test_cnn_module.py
imports: import pickle
  import akro
  import numpy
  import pytest
  import torch
  import torch.nn
  from garage import InOutSpec
  from garage.torch.modules import CNNModule
  from garage.torch.modules.cnn_module import _check_spec
label_desc: Use image node_id 1 for calling a global function with example usage: test_set_output_size(kernel_sizes, hidden_channels, strides, pool_shape, pool_stride) without return types