cuda.cuh

typedef struct {
    int rows;
    int cols;
    double* data;
} Matrix;

Matrix* matrix_create(int rows, int cols);

Matrix* matrix_from_shape(Matrix* m);

Matrix* matrix_from_data(int rows, int cols, double* data);

void matrix_free(Matrix* m);

Matrix* matrix_dot(Matrix* a, Matrix* b);

Matrix* matrix_transpose(Matrix* m);

Matrix* matrix_add(Matrix* a, Matrix* b);

Matrix* matrix_sub(Matrix* a, Matrix* b);

Matrix* matrix_mul(Matrix* a, Matrix* b);

Matrix* matrix_divf(Matrix* a, double f);

Matrix* matrix_subf(Matrix* m, double f);

Matrix* matrix_mulf(Matrix* m, double f);

double matrix_sum(Matrix* m);

void matrix_print(char* label, Matrix* m, int y, int x);

Matrix* matrix_rand(int rows, int cols);

Matrix* matrix_relu(Matrix* m);

Matrix* matrix_softmax(Matrix* m);

Matrix* matrix_one_hot(const double* Y, int len);

Matrix* matrix_cols(Matrix* m, int start, int end);