@Test
public void testCNNInputSetupMNIST() throws Exception {
INDArray input = getMnistData();
Layer layer = getMNISTConfig();
layer.activate(input);
assertEquals(input, layer.input());
assertArrayEquals(input.shape(), layer.input().shape());
}
@Test
public void testCNNInputSetup() throws Exception {
INDArray input = getMnistData();
int[] stride = new int[] {3, 3};
int[] padding = new int[] {1, 1};
Layer layer = getCNNConfig(nChannelsIn, depth, kernelSize, stride, padding);
layer.activate(input);
assertEquals(input, layer.input());
assertEquals(input.shape(), layer.input().shape());
}
