How to best concatenate 2 frames for an NN

I need to grab 2 frames, and then send to a NN with them concatenated as 1 frame.

Is there a low compute way to do this on depthai?

shape = (3, 720, 720)

X = torch.ones(shape, dtype=torch.float32)

path = Path("out/")

path.mkdir(parents=True, exist_ok=True)

onnx_file = "out/diff.onnx"

print(f"Writing to {onnx_file}")

torch.onnx.export(

Model(),

(X, X),

onnx_file,

opset_version=12,

do_constant_folding=True,

input_names = ['img1', 'img2'], # Optional

output_names = ['output'], # Optional

)

and how it compares to:

why is one example (X, X) and one is (X, X, X)?

Perfect thank you

AdamPolak one model takes 3 different (300x300) streams, the other 2 different 720x720 streams.