Merge pull request #5 from duprijil/to_pull

optimised code

Author: duprijil

MNIST_API_impl.c

@@ -160,6 +160,7 @@ uint32_t* mnistapi_getRandomTestImage(void) {
 
 
 uint32_t* mnistapi_getRandomTrainImageByNumber(uint32_t number) {
+    if (number > 9) return NULL;
     uint32_t randNumber = 0;
     uint32_t* image = (uint32_t*)malloc(sizeof(uint32_t) * IMAGE_SIZE);
     register uint32_t j = 0;
@@ -176,6 +177,7 @@ uint32_t* mnistapi_getRandomTrainImageByNumber(uint32_t number) {
 }
 
 uint32_t* mnistapi_getRandomTestImageByNumber(uint32_t number) {
+    if (number > 9) return NULL;
     uint32_t randNumber = 0;
     uint32_t* image = (uint32_t*)malloc(sizeof(uint32_t) * IMAGE_SIZE);
     register uint32_t j = 0;