tensors.h

Go to the documentation of this file.

 
#ifndef TENSORS_H
#define TENSORS_H
 
#include <stdbool.h>
#include <stdlib.h>
 
#define vector(TYPE, MACRO_LENGTH)                                          \
    ({                                                                      \
        TYPE* VECTOR_RESULT = (TYPE*)malloc((MACRO_LENGTH) * sizeof(TYPE)); \
        if (VECTOR_RESULT == NULL)                                          \
        {                                                                   \
            exit(EXIT_FAILURE);                                             \
        }                                                                   \
        VECTOR_RESULT;                                                      \
    })
 
#define zeros_vector(TYPE, MACRO_LENGTH)                                    \
    ({                                                                      \
        TYPE* VECTOR_RESULT = (TYPE*)calloc((MACRO_LENGTH), sizeof(TYPE));  \
        if (VECTOR_RESULT == NULL)                                          \
        {                                                                   \
            exit(EXIT_FAILURE);                                             \
        }                                                                   \
        VECTOR_RESULT;                                                      \
    })
 
#define matrix(TYPE, MACRO_WIDTH, MACRO_HEIGHT)                             \
    ({                                                                      \
        TYPE** MATRIX_RESULT = vector(TYPE*, (MACRO_WIDTH));                \
        for (int MACRO_i = 0; MACRO_i < (MACRO_WIDTH); MACRO_i++)           \
        {                                                                   \
            MATRIX_RESULT[MACRO_i] = vector(TYPE, (MACRO_HEIGHT));          \
        }                                                                   \
        MATRIX_RESULT;                                                      \
    })
 
#define zeros_matrix(TYPE, MACRO_WIDTH, MACRO_HEIGHT)                       \
    ({                                                                      \
        TYPE** MATRIX_RESULT = zeros_vector(TYPE*, (MACRO_WIDTH));          \
        for (int MACRO_i = 0; MACRO_i < (MACRO_WIDTH); MACRO_i++)           \
        {                                                                   \
            MATRIX_RESULT[MACRO_i] = zeros_vector(TYPE, (MACRO_HEIGHT));    \
        }                                                                   \
        MATRIX_RESULT;                                                      \
    })
 
#define free_matrix(MATRIX, MACRO_WIDTH)                                    \
    ({                                                                      \
        for (int MACRO_i = 0; MACRO_i < (MACRO_WIDTH); MACRO_i++)           \
        {                                                                   \
            free(MATRIX[MACRO_i]);                                          \
        }                                                                   \
        free(MATRIX);                                                       \
    })
 
#endif /* TENSORS_H */