devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Implement exclusive scan option

Browse Source
This commit is contained in:
Igor Babuschkin
2016-06-14 19:44:07 +01:00
parent 4c61f00838
commit c4d10e921f
3 changed files with 46 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
unsupported/test/cxx11_tensor_scan.cpp
View File

@@ -38,6 +38,30 @@ static void test_1d_scan()
}
}
template <int DataLayout, typename Type=float>
static void test_1d_inclusive_scan()
{
int size = 50;
Tensor<Type, 1, DataLayout> tensor(size);
tensor.setRandom();
Tensor<Type, 1, DataLayout> result = tensor.cumsum(0, true);
VERIFY_IS_EQUAL(tensor.dimension(0), result.dimension(0));
float accum = 0;
for (int i = 0; i < size; i++) {
VERIFY_IS_EQUAL(result(i), accum);
accum += tensor(i);
}
accum = 1;
result = tensor.cumprod(0, true);
for (int i = 0; i < size; i++) {
VERIFY_IS_EQUAL(result(i), accum);
accum *= tensor(i);
}
}
template <int DataLayout, typename Type=float>
static void test_4d_scan()
{