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

Use select ternary op in tensor select evaulator

Browse Source
This commit is contained in:
Charles Schlosser
2023-04-18 20:52:16 +00:00
committed by Rasmus Munk Larsen
parent 2b954be663
commit e2bbf496f6
3 changed files with 72 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
unsupported/test/cxx11_tensor_expr.cpp
View File

@@ -280,6 +280,8 @@ static void test_type_casting()
static void test_select()
{
using TypedGTOp = internal::scalar_cmp_op<float, float, internal::cmp_GT, true>;
Tensor<float, 3> selector(2,3,7);
Tensor<float, 3> mat1(2,3,7);
Tensor<float, 3> mat2(2,3,7);
@@ -288,6 +290,8 @@ static void test_select()
selector.setRandom();
mat1.setRandom();
mat2.setRandom();
// test select with a boolean condition
result = (selector > selector.constant(0.5f)).select(mat1, mat2);
for (int i = 0; i < 2; ++i) {
@@ -297,6 +301,18 @@ static void test_select()
}
}
}
// test select with a typed condition
result = selector.binaryExpr(selector.constant(0.5f), TypedGTOp()).select(mat1, mat2);
for (int i = 0; i < 2; ++i) {
for (int j = 0; j < 3; ++j) {
for (int k = 0; k < 7; ++k) {
VERIFY_IS_APPROX(result(i, j, k), (selector(i, j, k) > 0.5f) ? mat1(i, j, k) : mat2(i, j, k));
}
}
}
}
template <typename Scalar>