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

Added tensor addition.

Browse Source
This commit is contained in:
Elias Kohout
2023-02-27 14:45:21 +01:00
parent 3cb2f272eb
commit dbf11ef652
2 changed files with 19 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
tensor.c
View File

@@ -130,11 +130,26 @@ int tensor_init_zero(tensor t, int dimension, const int *size)
int tensor_init_rand(tensor t, int dimension, const int *size, int max) int tensor_init_rand(tensor t, int dimension, const int *size, int max)
{ {
int i; int i;
srand(time(NULL)); static int last_seed;
last_seed += time(NULL) * 200 + rand();
srand(last_seed);
if(!_tensor_set_size(t, size, dimension)) return 0; if(!_tensor_set_size(t, size, dimension)) return 0;
for(i = 0; i < t->num_elem; i++) { for(i = 0; i < t->num_elem; i++) {
t->elements[i] = (dtype) (rand() % max); t->elements[i] = (dtype) ((double) rand() / RAND_MAX * max);
}
return 1;
}
int tensor_add(tensor t1, const tensor t2)
{
int i;
if(t1->dimension != t2->dimension) return 0;
for(i = 0; i < t1->dimension; i++) {
if(t1->size[i] != t2->size[i]) return 0;
}
for(i = 0; i < t1->num_elem; i++) {
t1->elements[i] += t2->elements[i];
} }
return 1; return 1;
} }
@@ -142,8 +157,6 @@ int tensor_init_rand(tensor t, int dimension, const int *size, int max)
void tensor_for_each_elem(tensor t, dtype (*func)(dtype)) void tensor_for_each_elem(tensor t, dtype (*func)(dtype))
{ {
int i; int i;
srand(time(NULL));
for(i = 0; i < t->num_elem; i++) { for(i = 0; i < t->num_elem; i++) {
t->elements[i] = func(t->elements[i]); t->elements[i] = func(t->elements[i]);
} }

2
tensor.h
View File

@@ -33,6 +33,8 @@ int tensor_init_one(tensor t, int dimension, const int *size);
int tensor_init_zero(tensor t, int dimension, const int *size); int tensor_init_zero(tensor t, int dimension, const int *size);
int tensor_init_rand(tensor t, int dimension, const int *size, int max); int tensor_init_rand(tensor t, int dimension, const int *size, int max);
int tensor_add(tensor t1, const tensor t2);
void tensor_for_each_elem(tensor t, dtype (*func)(dtype)); void tensor_for_each_elem(tensor t, dtype (*func)(dtype));
void tensor_print(const tensor t); void tensor_print(const tensor t);