Fix Concatenate on different axes
- Keras Concatenate layer assumes
axis=0
is the batch dimension, so if we are parsing that directly, we need to consider that, i.e.- for 2D concatenation,
axis = 2
oraxis = -1
, means we useconcatenate2d_1
; otherwise we useconcatenate2d_0
- for 3D concatenation,
axis = 3
oraxis = -1
, means we useconcatenate3d_2
;axis = 2
oraxis = -2
means we useconcatenate3d_1
; otherwise we useconcatenate3d_0
- for 2D concatenation,
- Also fix shape inference