@Test public void canConcat() { IntegerVertex A = new UniformIntVertex(0, 1); A.setValue(IntegerTensor.ones(2, 2)); IntegerVertex B = new UniformIntVertex(0, 1); B.setValue(IntegerTensor.ones(2, 2)); IntegerVertex concatDimZero = IntegerVertex.concat(0, A, B); assertArrayEquals(concatDimZero.getShape(), new long[]{4, 2}); IntegerVertex concatDimOne = IntegerVertex.concat(1, A, B); assertArrayEquals(concatDimOne.getShape(), new long[]{2, 4}); }