code: def test_conv2d_output_shape(
      input_dim, kernel_size, stride, padding, dilation
  ):
      h, w = conv2d_output_shape(
          input_dim,
          kernel_size=kernel_size,
          stride=stride,
          pad=padding,
          dilation=dilation,
      )
      conv = torch.nn.Conv2d(
          1,
          1,
          kernel_size,
          stride=stride,
          padding=padding,
          dilation=dilation,
      )
      x = torch.rand(1, 1, *input_dim)
      assert conv(x).shape[2:] == (h, w)
  
imports: import pytest
  import torch
  from espnet2.enh.layers.conv_utils import conv2d_output_shape, convtransp2d_output_shape
label_desc: Use image node_id 1 for calling a global function with example usage: test_conv2d_output_shape(input_dim, kernel_size, stride, padding, dilation) without return types