val convConfig = Conv2DConfig.builder()
                        .kH(kernelShape[0].toLong())
                        .kW(kernelShape[1].toLong())
                        .sH(strides[0])
                        .sW(strides[1])