Implemented the new elementwise functions.
This commit is contained in:
10
tensorfunc.h
10
tensorfunc.h
@@ -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);
|
||||
|
||||
Reference in New Issue
Block a user