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

Added more tests.

Browse Source
This commit is contained in:
Elias Kohout
2023-09-04 16:41:07 +02:00
parent 3543700b01
commit 8c2906765f
9 changed files with 170 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
tensoriter.h Normal file
View File

@@ -0,0 +1,24 @@
#ifndef _TENSORITERATOR_H_
#define _TENSORITERATOR_H_
#include "tensor.h"
typedef struct _tensor_scalar_iterator {
dtype *current;
uint32_t length;
} * tensoriter_scalar;
tensoriter_scalar tensoriter_scalar_create(tensor t);
void tensoriter_scalar_destroy(tensoriter_scalar it);
bool tensoriter_scalar_next(tensoriter_scalar it);
dtype *tensoriter_scalar_get(tensoriter_scalar it);
void tensoriter_scalar_map(tensoriter_scalar it, dtype (*func)(dtype));
void tensoriter_scalar_map_add(tensoriter_scalar it, dtype scalar);
void tensoriter_scalar_map_sub(tensoriter_scalar it, dtype scalar);
void tensoriter_scalar_map_mul(tensoriter_scalar it, dtype scalar);
void tensoriter_scalar_map_div(tensoriter_scalar it, dtype scalar);
#endif // _TENSORITERATOR_H_