Implemented the new elementwise functions.

This commit is contained in:
2023-09-15 19:22:43 +02:00
parent 79ed621ba2
commit 5ec1d9e73a
4 changed files with 85 additions and 28 deletions

View File

@@ -10,17 +10,17 @@ tensor tensor_map(const tensor t, dtype (*func)(dtype));
void tensor_map_inplace(tensor t, dtype (*func)(dtype));
tensor tensor_combine(const tensor t1, const tensor t2, dtype (*func)(dtype, dtype));
void tensor_combine_inplace(tensor t1, const tensor t2, dtype (*func)(dtype, dtype));
bool tensor_combine_inplace(tensor t1, const tensor t2, dtype (*func)(dtype, dtype));
void tensor_add_scalar(tensor t, dtype scalar);
void tensor_sub_scalar(tensor t, dtype scalar);
void tensor_mul_scalar(tensor t, dtype scalar);
void tensor_div_scalar(tensor t, dtype scalar);
void tensor_add_inplace(tensor t1, const tensor t2);
void tensor_sub_inplace(tensor t1, const tensor t2);
void tensor_mul_inplace(tensor t1, const tensor t2);
void tensor_div_inplace(tensor t1, const tensor t2);
bool tensor_add_inplace(tensor t1, const tensor t2);
bool tensor_sub_inplace(tensor t1, const tensor t2);
bool tensor_mul_inplace(tensor t1, const tensor t2);
bool tensor_div_inplace(tensor t1, const tensor t2);
tensor tensor_add(const tensor t1, const tensor t2);
tensor tensor_sub(const tensor t1, const tensor t2);