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

Clang-format tests, examples, libraries, benchmarks, etc.

Browse Source
This commit is contained in:
Antonio Sánchez
2023-12-05 21:22:55 +00:00
committed by Rasmus Munk Larsen
parent 3252ecc7a4
commit 46e9cdb7fe
876 changed files with 33453 additions and 37795 deletions
Download Patch File Download Diff File Expand all files Collapse all files

103
unsupported/test/cxx11_tensor_ref.cpp
View File

@@ -11,11 +11,10 @@
#include <Eigen/CXX11/Tensor>
using Eigen::Tensor;
using Eigen::RowMajor;
using Eigen::Tensor;
static void test_simple_lvalue_ref()
{
static void test_simple_lvalue_ref() {
Tensor<int, 1> input(6);
input.setRandom();
@@ -40,13 +39,11 @@ static void test_simple_lvalue_ref()
ref4.coeffRef(i) = -i * 2;
}
for (int i = 0; i < 6; ++i) {
VERIFY_IS_EQUAL(input(i), -i*2);
VERIFY_IS_EQUAL(input(i), -i * 2);
}
}
static void test_simple_rvalue_ref()
{
static void test_simple_rvalue_ref() {
Tensor<int, 1> input1(6);
input1.setRandom();
Tensor<int, 1> input2(6);
@@ -66,10 +63,8 @@ static void test_simple_rvalue_ref()
}
}
static void test_multiple_dims()
{
Tensor<float, 3> input(3,5,7);
static void test_multiple_dims() {
Tensor<float, 3> input(3, 5, 7);
input.setRandom();
TensorRef<Tensor<float, 3>> ref(input);
@@ -81,44 +76,40 @@ static void test_multiple_dims()
for (int i = 0; i < 3; ++i) {
for (int j = 0; j < 5; ++j) {
for (int k = 0; k < 7; ++k) {
VERIFY_IS_EQUAL(ref(i,j,k), input(i,j,k));
VERIFY_IS_EQUAL(ref(i, j, k), input(i, j, k));
}
}
}
}
static void test_slice()
{
Tensor<float, 5> tensor(2,3,5,7,11);
static void test_slice() {
Tensor<float, 5> tensor(2, 3, 5, 7, 11);
tensor.setRandom();
Eigen::DSizes<ptrdiff_t, 5> indices(1,2,3,4,5);
Eigen::DSizes<ptrdiff_t, 5> sizes(1,1,1,1,1);
Eigen::DSizes<ptrdiff_t, 5> indices(1, 2, 3, 4, 5);
Eigen::DSizes<ptrdiff_t, 5> sizes(1, 1, 1, 1, 1);
TensorRef<Tensor<float, 5>> slice = tensor.slice(indices, sizes);
VERIFY_IS_EQUAL(slice(0,0,0,0,0), tensor(1,2,3,4,5));
VERIFY_IS_EQUAL(slice(0, 0, 0, 0, 0), tensor(1, 2, 3, 4, 5));
Eigen::DSizes<ptrdiff_t, 5> indices2(1,1,3,4,5);
Eigen::DSizes<ptrdiff_t, 5> sizes2(1,1,2,2,3);
Eigen::DSizes<ptrdiff_t, 5> indices2(1, 1, 3, 4, 5);
Eigen::DSizes<ptrdiff_t, 5> sizes2(1, 1, 2, 2, 3);
slice = tensor.slice(indices2, sizes2);
for (int i = 0; i < 2; ++i) {
for (int j = 0; j < 2; ++j) {
for (int k = 0; k < 3; ++k) {
VERIFY_IS_EQUAL(slice(0,0,i,j,k), tensor(1,1,3+i,4+j,5+k));
VERIFY_IS_EQUAL(slice(0, 0, i, j, k), tensor(1, 1, 3 + i, 4 + j, 5 + k));
}
}
}
Eigen::DSizes<ptrdiff_t, 5> indices3(0,0,0,0,0);
Eigen::DSizes<ptrdiff_t, 5> sizes3(2,3,1,1,1);
Eigen::DSizes<ptrdiff_t, 5> indices3(0, 0, 0, 0, 0);
Eigen::DSizes<ptrdiff_t, 5> sizes3(2, 3, 1, 1, 1);
slice = tensor.slice(indices3, sizes3);
VERIFY_IS_EQUAL(slice.data(), tensor.data());
}
static void test_ref_of_ref()
{
Tensor<float, 3> input(3,5,7);
static void test_ref_of_ref() {
Tensor<float, 3> input(3, 5, 7);
input.setRandom();
TensorRef<Tensor<float, 3>> ref(input);
@@ -139,33 +130,31 @@ static void test_ref_of_ref()
for (int i = 0; i < 3; ++i) {
for (int j = 0; j < 5; ++j) {
for (int k = 0; k < 7; ++k) {
VERIFY_IS_EQUAL(ref_of_ref(i,j,k), input(i,j,k));
VERIFY_IS_EQUAL(ref_of_ref2(i,j,k), input(i,j,k));
}
VERIFY_IS_EQUAL(ref_of_ref(i, j, k), input(i, j, k));
VERIFY_IS_EQUAL(ref_of_ref2(i, j, k), input(i, j, k));
}
}
}
}
static void test_ref_in_expr()
{
Tensor<float, 3> input(3,5,7);
static void test_ref_in_expr() {
Tensor<float, 3> input(3, 5, 7);
input.setRandom();
TensorRef<Tensor<float, 3>> input_ref(input);
Tensor<float, 3> result(3,5,7);
Tensor<float, 3> result(3, 5, 7);
result.setRandom();
TensorRef<Tensor<float, 3>> result_ref(result);
Tensor<float, 3> bias(3,5,7);
Tensor<float, 3> bias(3, 5, 7);
bias.setRandom();
result_ref = input_ref + bias;
for (int i = 0; i < 3; ++i) {
for (int j = 0; j < 5; ++j) {
for (int k = 0; k < 7; ++k) {
VERIFY_IS_EQUAL(result_ref(i,j,k), input(i,j,k) + bias(i,j,k));
VERIFY_IS_NOT_EQUAL(result(i,j,k), input(i,j,k) + bias(i,j,k));
VERIFY_IS_EQUAL(result_ref(i, j, k), input(i, j, k) + bias(i, j, k));
VERIFY_IS_NOT_EQUAL(result(i, j, k), input(i, j, k) + bias(i, j, k));
}
}
}
@@ -174,16 +163,14 @@ static void test_ref_in_expr()
for (int i = 0; i < 3; ++i) {
for (int j = 0; j < 5; ++j) {
for (int k = 0; k < 7; ++k) {
VERIFY_IS_EQUAL(result(i,j,k), input(i,j,k) + bias(i,j,k));
VERIFY_IS_EQUAL(result(i, j, k), input(i, j, k) + bias(i, j, k));
}
}
}
}
static void test_coeff_ref()
{
Tensor<float, 5> tensor(2,3,5,7,11);
static void test_coeff_ref() {
Tensor<float, 5> tensor(2, 3, 5, 7, 11);
tensor.setRandom();
Tensor<float, 5> original = tensor;
@@ -191,42 +178,40 @@ static void test_coeff_ref()
slice.coeffRef(0, 0, 0, 0) = 1.0f;
slice.coeffRef(1, 0, 0, 0) += 2.0f;
VERIFY_IS_EQUAL(tensor(0,0,0,0,7), 1.0f);
VERIFY_IS_EQUAL(tensor(1,0,0,0,7), original(1,0,0,0,7) + 2.0f);
VERIFY_IS_EQUAL(tensor(0, 0, 0, 0, 7), 1.0f);
VERIFY_IS_EQUAL(tensor(1, 0, 0, 0, 7), original(1, 0, 0, 0, 7) + 2.0f);
}
static void test_nested_ops_with_ref()
{
static void test_nested_ops_with_ref() {
Tensor<float, 4> t(2, 3, 5, 7);
t.setRandom();
TensorMap<Tensor<const float, 4> > m(t.data(), 2, 3, 5, 7);
TensorMap<Tensor<const float, 4>> m(t.data(), 2, 3, 5, 7);
array<std::pair<ptrdiff_t, ptrdiff_t>, 4> paddings;
paddings[0] = std::make_pair(0, 0);
paddings[1] = std::make_pair(2, 1);
paddings[2] = std::make_pair(3, 4);
paddings[3] = std::make_pair(0, 0);
DSizes<Eigen::DenseIndex, 4> shuffle_dims(0, 1, 2, 3);
TensorRef<Tensor<const float, 4> > ref(m.pad(paddings));
TensorRef<Tensor<const float, 4>> ref(m.pad(paddings));
array<std::pair<ptrdiff_t, ptrdiff_t>, 4> trivial;
trivial[0] = std::make_pair(0, 0);
trivial[1] = std::make_pair(0, 0);
trivial[2] = std::make_pair(0, 0);
trivial[3] = std::make_pair(0, 0);
Tensor<float, 4> padded = ref.shuffle(shuffle_dims).pad(trivial);
VERIFY_IS_EQUAL(padded.dimension(0), 2+0);
VERIFY_IS_EQUAL(padded.dimension(1), 3+3);
VERIFY_IS_EQUAL(padded.dimension(2), 5+7);
VERIFY_IS_EQUAL(padded.dimension(3), 7+0);
VERIFY_IS_EQUAL(padded.dimension(0), 2 + 0);
VERIFY_IS_EQUAL(padded.dimension(1), 3 + 3);
VERIFY_IS_EQUAL(padded.dimension(2), 5 + 7);
VERIFY_IS_EQUAL(padded.dimension(3), 7 + 0);
for (int i = 0; i < 2; ++i) {
for (int j = 0; j < 6; ++j) {
for (int k = 0; k < 12; ++k) {
for (int l = 0; l < 7; ++l) {
if (j >= 2 && j < 5 && k >= 3 && k < 8) {
VERIFY_IS_EQUAL(padded(i,j,k,l), t(i,j-2,k-3,l));
VERIFY_IS_EQUAL(padded(i, j, k, l), t(i, j - 2, k - 3, l));
} else {
VERIFY_IS_EQUAL(padded(i,j,k,l), 0.0f);
VERIFY_IS_EQUAL(padded(i, j, k, l), 0.0f);
}
}
}
@@ -234,9 +219,7 @@ static void test_nested_ops_with_ref()
}
}
EIGEN_DECLARE_TEST(cxx11_tensor_ref)
{
EIGEN_DECLARE_TEST(cxx11_tensor_ref) {
CALL_SUBTEST(test_simple_lvalue_ref());
CALL_SUBTEST(test_simple_rvalue_ref());
CALL_SUBTEST(test_multiple_dims());