From 30c459f24f8bb2068c8cb93960e7d4d165595a1a Mon Sep 17 00:00:00 2001 From: Elias Kohout Date: Fri, 24 Feb 2023 19:20:09 +0100 Subject: [PATCH] Added max argument to init_rand function. --- tensor.c | 4 ++-- tensor.h | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tensor.c b/tensor.c index c9c5068..ecf5afa 100644 --- a/tensor.c +++ b/tensor.c @@ -119,14 +119,14 @@ int tensor_init_zero(tensor t, int dimension, const int *size) 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; srand(time(NULL)); if(!_tensor_set_size(t, size, dimension)) return 0; for(i = 0; i < t->num_elem; i++) { - t->elements[i] = (t_type) rand(); + t->elements[i] = (t_type) (rand() % max); } return 1; } diff --git a/tensor.h b/tensor.h index 8c10292..7fb9473 100644 --- a/tensor.h +++ b/tensor.h @@ -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_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_print(const tensor t);