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

Add syntactic sugar to Eigen tensors to allow more natural syntax.

Browse Source 
Specifically, this enables expressions involving:

scalar + tensor
scalar * tensor
scalar / tensor
scalar - tensor
This commit is contained in:
Rasmus Munk Larsen
2016-06-02 12:41:28 -07:00
parent 6021c90fdf
commit 811aadbe00
2 changed files with 57 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
unsupported/test/cxx11_tensor_sugar.cpp
View File

@@ -33,7 +33,7 @@ static void test_comparison_sugar() {
}
static void test_scalar_sugar() {
static void test_scalar_sugar_add_mul() {
Tensor<float, 3> A(6, 7, 5);
Tensor<float, 3> B(6, 7, 5);
A.setRandom();
@@ -41,21 +41,41 @@ static void test_scalar_sugar() {
const float alpha = 0.43f;
const float beta = 0.21f;
const float gamma = 0.14f;
Tensor<float, 3> R = A * A.constant(alpha) + B * B.constant(beta);
Tensor<float, 3> S = A * alpha + B * beta;
Tensor<float, 3> R = A.constant(gamma) + A * A.constant(alpha) + B * B.constant(beta);
Tensor<float, 3> S = A * alpha + B * beta + gamma;
Tensor<float, 3> T = gamma + alpha * A + beta * B;
// TODO: add enough syntactic sugar to support this
// Tensor<float, 3> T = alpha * A + beta * B;
for (int i = 0; i < 6*7*5; ++i) {
VERIFY_IS_APPROX(R(i), S(i));
VERIFY_IS_APPROX(R(i), T(i));
}
}
static void test_scalar_sugar_sub_div() {
Tensor<float, 3> A(6, 7, 5);
Tensor<float, 3> B(6, 7, 5);
A.setRandom();
B.setRandom();
const float alpha = 0.43f;
const float beta = 0.21f;
const float gamma = 0.14f;
const float delta = 0.32f;
Tensor<float, 3> R = A.constant(gamma) - A / A.constant(alpha)
- B.constant(beta) / B - A.constant(delta);
Tensor<float, 3> S = gamma - A / alpha - beta / B - delta;
for (int i = 0; i < 6*7*5; ++i) {
VERIFY_IS_APPROX(R(i), S(i));
}
}
void test_cxx11_tensor_sugar()
{
CALL_SUBTEST(test_comparison_sugar());
CALL_SUBTEST(test_scalar_sugar());
CALL_SUBTEST(test_scalar_sugar_add_mul());
CALL_SUBTEST(test_scalar_sugar_sub_div());
}