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

Added iterators over scalars of tensors

Browse Source
This commit is contained in:
Elias Kohout
2023-05-07 16:58:49 +02:00
parent 78623d2e9d
commit 19c644f450
4 changed files with 112 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
tensor.c
View File

@@ -1,8 +1,5 @@
#include "tensor.h"
/* A helper for f.e. the tensor_add_scalar function */
dtype _tensor_scalar_wise_helper;
#include "tensoriterator.h"
tensor tensor_new(void)
{
@@ -189,41 +186,40 @@ uint8_t tensor_cpy(tensor t1, const tensor t2)
return 1;
}
dtype _tensor_dtype_add_helper(dtype d) { return DTYPE_ADD(d, _tensor_scalar_wise_helper); }
dtype _tensor_dtype_sub_helper(dtype d) { return DTYPE_SUB(d, _tensor_scalar_wise_helper); }
dtype _tensor_dtype_mul_helper(dtype d) { return DTYPE_MUL(d, _tensor_scalar_wise_helper); }
dtype _tensor_dtype_div_helper(dtype d) { return DTYPE_DIV(d, _tensor_scalar_wise_helper); }
void tensor_add_scalar(tensor t, dtype n)
{
assert(!tensor_is_empty(t));
_tensor_scalar_wise_helper = n;
tensor_map(t, &_tensor_dtype_add_helper);
tensoriter_scalar iter = tensoriter_scalar_create(t);
tensoriter_scalar_map_add(iter, n);
tensoriter_scalar_destroy(iter);
}
void tensor_sub_scalar(tensor t, dtype n)
{
assert(!tensor_is_empty(t));
_tensor_scalar_wise_helper = n;
tensor_map(t, &_tensor_dtype_sub_helper);
tensoriter_scalar iter = tensoriter_scalar_create(t);
tensoriter_scalar_map_sub(iter, n);
tensoriter_scalar_destroy(iter);
}
void tensor_mul_scalar(tensor t, dtype n)
{
assert(!tensor_is_empty(t));
_tensor_scalar_wise_helper = n;
tensor_map(t, &_tensor_dtype_mul_helper);
tensoriter_scalar iter = tensoriter_scalar_create(t);
tensoriter_scalar_map_mul(iter, n);
tensoriter_scalar_destroy(iter);
}
void tensor_div_scalar(tensor t, dtype n)
{
assert(!tensor_is_empty(t));
_tensor_scalar_wise_helper = n;
tensor_map(t, &_tensor_dtype_div_helper);
tensoriter_scalar iter = tensoriter_scalar_create(t);
tensoriter_scalar_map_div(iter, n);
tensoriter_scalar_destroy(iter);
}
uint8_t tensor_add_inplace(tensor t1, const tensor t2)
@@ -283,16 +279,6 @@ tensor tensor_sub(const tensor t1, const tensor t2)
}
void tensor_map(tensor t, dtype (*func)(dtype))
{
assert(!tensor_is_empty(t));
uint32_t i;
for(i = 0; i < t->num_elem; i++) {
t->elements[i] = func(t->elements[i]);
}
}
void tensor_print(const tensor t)
{
uint32_t i, j;