eliaskohout/ctensor
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Minor bug fixes.

Browse Source
This commit is contained in:
Elias Kohout
2023-03-09 18:24:17 +01:00
parent dbf11ef652
commit 7d22af5e85
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tensor.c
View File

@@ -40,10 +40,10 @@ int _tensor_set_size(tensor t, const int *size, int dim)
}
if(!_tensor_check_size(size, dim)) return 0;
/* Try allocating memory for the size array of the Tensor */
/* Try allocating memory for the size array of the tensor */
temp = realloc(t->size, dim * sizeof(int));
if(temp == NULL && dim != 0) return 0;
/* Try allocating memory for the Tensor */
/* Try allocating memory for the tensor */
t_temp = realloc(t->elements, num_elem * sizeof(dtype));
if(t_temp == NULL) {
/* Revert to before the function call and return */
@@ -70,7 +70,10 @@ int tensor_set(tensor t, const int *index, dtype val)
{
int i, offset = 0;
if(tensor_is_empty(t)) return 0;
if(t->dimension == 0) return t->elements[0] = val;
if(t->dimension == 0) {
t->elements[0] = val;
return 1;
}
for(i = 0; i < t->dimension - 1; i++) {
if(t->size[i] <= index[i]) return 0;
@@ -181,11 +184,11 @@ void tensor_print(const tensor t)
if(t->dimension == 0) {
/* Skalar */
/* scalar */
printf(PRINT_STRING, t->elements[0]);
putchar('\n');
} else if (t->dimension == 1) {
/* Spaltenvektor */
/* column vector */
if(t->size[0] == 1) {
putchar('(');
printf(PRINT_STRING, t->elements[0]);
@@ -204,7 +207,7 @@ void tensor_print(const tensor t)
printf("/\n");
}
} else if (t->dimension == 2) {
/* Matix */
/* matix */
indx = malloc(sizeof(int) * 2);
if(t->size[0] == 1) {
putchar('(');