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

started implementation

Browse Source
This commit is contained in:
Elias Kohout
2023-09-15 18:22:37 +02:00
parent b9ecc03504
commit 6be69e8a6c
1 changed files with 14 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tensorfunc.c
View File

@@ -86,6 +86,12 @@ void tensor_combine_inplace(tensor t1, const tensor t2, dtype (*func)(dtype, dty
} }
dtype _dtype_scalar_helper;
dtype _tensor_add_scalar_helper(dtype x) { return DTYPE_ADD(x, _dtype_scalar_helper); }
dtype _tensor_sub_scalar_helper(dtype x) { return DTYPE_SUB(x, _dtype_scalar_helper); }
dtype _tensor_mul_scalar_helper(dtype x) { return DTYPE_MUL(x, _dtype_scalar_helper); }
dtype _tensor_div_scalar_helper(dtype x) { return DTYPE_DIV(x, _dtype_scalar_helper); }
void tensor_add_scalar(tensor t, dtype scalar) void tensor_add_scalar(tensor t, dtype scalar)
{ {
/* Adds a fixed scalar value to all the values of a tensor. /* Adds a fixed scalar value to all the values of a tensor.
@@ -93,7 +99,8 @@ void tensor_add_scalar(tensor t, dtype scalar)
* @param t The tensor to operate on * @param t The tensor to operate on
* @param scalar The value to add * @param scalar The value to add
*/ */
// TODO _dtype_scalar_helper = scalar;
tensor_map_inplace(t, &_tensor_add_scalar_helper);
} }
void tensor_sub_scalar(tensor t, dtype scalar) void tensor_sub_scalar(tensor t, dtype scalar)
@@ -103,7 +110,8 @@ void tensor_sub_scalar(tensor t, dtype scalar)
* @param t The tensor to operate on * @param t The tensor to operate on
* @param scalar The value to subtract * @param scalar The value to subtract
*/ */
// TODO _dtype_scalar_helper = scalar;
tensor_map_inplace(t, &_tensor_sub_scalar_helper);
} }
void tensor_mul_scalar(tensor t, dtype scalar) void tensor_mul_scalar(tensor t, dtype scalar)
@@ -113,7 +121,8 @@ void tensor_mul_scalar(tensor t, dtype scalar)
* @param t The tensor to operate on * @param t The tensor to operate on
* @param scalar The value to multiply * @param scalar The value to multiply
*/ */
// TODO _dtype_scalar_helper = scalar;
tensor_map_inplace(t, &_tensor_mul_scalar_helper);
} }
void tensor_div_scalar(tensor t, dtype scalar) void tensor_div_scalar(tensor t, dtype scalar)
@@ -123,7 +132,8 @@ void tensor_div_scalar(tensor t, dtype scalar)
* @param t The tensor to operate on * @param t The tensor to operate on
* @param scalar The value to divide by * @param scalar The value to divide by
*/ */
// TODO _dtype_scalar_helper = scalar;
tensor_map_inplace(t, &_tensor_mul_scalar_helper);
} }
void tensor_add_inplace(tensor t1, const tensor t2) void tensor_add_inplace(tensor t1, const tensor t2)