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

Added tensor_add/tensor_sub functions.

Browse Source
This commit is contained in:
Elias Kohout
2023-04-11 20:55:16 +02:00
parent 721ac75389
commit fa172de1eb
1 changed files with 25 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
tensor.c
View File

@@ -187,7 +187,6 @@ int tensor_cpy(tensor t1, const tensor t2)
return 1; return 1;
} }
dtype _tensor_dtype_add_helper(dtype d) { return DTYPE_ADD(d, _tensor_scalar_wise_helper); } 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_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_mul_helper(dtype d) { return DTYPE_MUL(d, _tensor_scalar_wise_helper); }
@@ -257,6 +256,31 @@ int tensor_sub_inplace(tensor t1, const tensor t2)
return 1; return 1;
} }
tensor tensor_add(const tensor t1, const tensor t2)
{
assert(!tensor_is_empty(t1));
assert(!tensor_is_empty(t2));
tensor t3 = tensor_new();
if(t3 == NULL) return NULL;
if (!tensor_cpy(t3, t1)) return NULL;
if (!tensor_add_inplace(t3, t2)) return NULL;
return t3;
}
tensor tensor_sub(const tensor t1, const tensor t2)
{
assert(!tensor_is_empty(t1));
assert(!tensor_is_empty(t2));
tensor t3 = tensor_new();
if(t3 == NULL) return NULL;
if (!tensor_cpy(t3, t1)) return NULL;
if (!tensor_sub_inplace(t3, t2)) return NULL;
return t3;
}
void tensor_map(tensor t, dtype (*func)(dtype)) void tensor_map(tensor t, dtype (*func)(dtype))
{ {
assert(!tensor_is_empty(t)); assert(!tensor_is_empty(t));