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