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

Added max argument to init_rand function.

Browse Source
This commit is contained in:
Elias Kohout
2023-02-24 19:20:09 +01:00
parent fdfe67d83e
commit 30c459f24f
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tensor.c
View File

@@ -119,14 +119,14 @@ int tensor_init_zero(tensor t, int dimension, const int *size)
return 1; return 1;
} }
int tensor_init_rand(tensor t, int dimension, const int *size) int tensor_init_rand(tensor t, int dimension, const int *size, int max)
{ {
int i; int i;
srand(time(NULL)); srand(time(NULL));
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] = (t_type) rand(); t->elements[i] = (t_type) (rand() % max);
} }
return 1; return 1;
} }

2
tensor.h
View File

@@ -31,7 +31,7 @@ t_type tensor_get(const tensor t, const int *index, int *success);
int tensor_init_one(tensor t, int dimension, const int *size); 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 tensor_init_rand(tensor t, int dimension, const int *size, int max);
void tensor_for_each_elem(tensor t, t_type (*func)(t_type)); void tensor_for_each_elem(tensor t, t_type (*func)(t_type));
void tensor_print(const tensor t); void tensor_print(const tensor t);