code: def forward(self, x, y):
      # In this test the shape will be (batch, 3, 10, 10), but since we test with channels_last data format, this
      # will be equivalent to the Keras default of concatenation along the last axis (axis=-1)
      return torch.cat([x, y], dim=1)
  
file_path: hls4ml/test/pytest/test_merge_pytorch.py
imports: from pathlib import Path
  import numpy
  import pytest
  import torch
  import torch.nn
  import hls4ml
label_desc: Use image node_id 2 for calling the ConcatModule obj's underlying member method code with example usage: obj.forward(x, y) and returns: torch