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

Added scalar math fuctions.

Browse Source
This commit is contained in:
Elias Kohout
2023-03-18 20:52:26 +01:00
parent b3490d36bf
commit 52d240fa98
2 changed files with 57 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

64
tensor.c
View File

@@ -170,6 +170,58 @@ int tensor_init_rand(tensor t, int dimension, const int *size, int max)
return 1;
}
int tensor_cpy(tensor t1, const tensor t2)
{
assert(!tensor_is_empty(t2));
int i;
if(!_tensor_set_size(t1, t2->size, t2->dimension)) return 0;
for(i = 0; i < t2->num_elem; i++) {
t1->elements[i] = t2->elements[i];
}
return 1;
}
void tensor_add_scalar(tensor t, dtype n)
{
assert(!tensor_is_empty(t));
int i;
for(i = 0; i < t->num_elem; i++) {
t->elements[i] += n;
}
}
void tensor_sub_scalar(tensor t, dtype n)
{
assert(!tensor_is_empty(t));
int i;
for(i = 0; i < t->num_elem; i++) {
t->elements[i] -= n;
}
}
void tensor_mult_scalar(tensor t, dtype n)
{
assert(!tensor_is_empty(t));
int i;
for(i = 0; i < t->num_elem; i++) {
t->elements[i] *= n;
}
}
void tensor_div_scalar(tensor t, dtype n)
{
assert(!tensor_is_empty(t));
int i;
for(i = 0; i < t->num_elem; i++) {
t->elements[i] /= n;
}
}
int tensor_add(tensor t1, const tensor t2)
{
assert(!tensor_is_empty(t1));
@@ -196,18 +248,6 @@ void tensor_for_each_elem(tensor t, dtype (*func)(dtype))
}
}
int tensor_cpy(tensor t1, const tensor t2)
{
assert(!tensor_is_empty(t2));
int i;
if(!_tensor_set_size(t1, t2->size, t2->dimension)) return 0;
for(i = 0; i < t2->num_elem; i++) {
t1->elements[i] = t2->elements[i];
}
return 1;
}
void tensor_print(const tensor t)
{
int i, j;