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

145
unsupported/test/cxx11_tensor_assign.cpp
View File

@@ -11,19 +11,24 @@
#include <Eigen/CXX11/Tensor>
using Eigen::Tensor;
using Eigen::RowMajor;
using Eigen::Tensor;
static void test_1d()
{
static void test_1d() {
Tensor<int, 1> vec1(6);
Tensor<int, 1, RowMajor> vec2(6);
vec1(0) = 4; vec2(0) = 0;
vec1(1) = 8; vec2(1) = 1;
vec1(2) = 15; vec2(2) = 2;
vec1(3) = 16; vec2(3) = 3;
vec1(4) = 23; vec2(4) = 4;
vec1(5) = 42; vec2(5) = 5;
vec1(0) = 4;
vec2(0) = 0;
vec1(1) = 8;
vec2(1) = 1;
vec1(2) = 15;
vec2(2) = 2;
vec1(3) = 16;
vec2(3) = 3;
vec1(4) = 23;
vec2(4) = 4;
vec1(5) = 42;
vec2(5) = 5;
int col_major[6] = {0};
int row_major[6] = {0};
@@ -67,24 +72,23 @@ static void test_1d()
VERIFY_IS_EQUAL(vec2(5), 5);
}
static void test_2d()
{
Tensor<int, 2> mat1(2,3);
Tensor<int, 2, RowMajor> mat2(2,3);
static void test_2d() {
Tensor<int, 2> mat1(2, 3);
Tensor<int, 2, RowMajor> mat2(2, 3);
mat1(0,0) = 0;
mat1(0,1) = 1;
mat1(0,2) = 2;
mat1(1,0) = 3;
mat1(1,1) = 4;
mat1(1,2) = 5;
mat1(0, 0) = 0;
mat1(0, 1) = 1;
mat1(0, 2) = 2;
mat1(1, 0) = 3;
mat1(1, 1) = 4;
mat1(1, 2) = 5;
mat2(0,0) = 0;
mat2(0,1) = 1;
mat2(0,2) = 2;
mat2(1,0) = 3;
mat2(1,1) = 4;
mat2(1,2) = 5;
mat2(0, 0) = 0;
mat2(0, 1) = 1;
mat2(0, 2) = 2;
mat2(1, 0) = 3;
mat2(1, 1) = 4;
mat2(1, 2) = 5;
int col_major[6] = {0};
int row_major[6] = {0};
@@ -94,58 +98,57 @@ static void test_2d()
mat3 = mat1;
mat4 = mat2;
VERIFY_IS_EQUAL(mat3(0,0), 0);
VERIFY_IS_EQUAL(mat3(0,1), 1);
VERIFY_IS_EQUAL(mat3(0,2), 2);
VERIFY_IS_EQUAL(mat3(1,0), 3);
VERIFY_IS_EQUAL(mat3(1,1), 4);
VERIFY_IS_EQUAL(mat3(1,2), 5);
VERIFY_IS_EQUAL(mat3(0, 0), 0);
VERIFY_IS_EQUAL(mat3(0, 1), 1);
VERIFY_IS_EQUAL(mat3(0, 2), 2);
VERIFY_IS_EQUAL(mat3(1, 0), 3);
VERIFY_IS_EQUAL(mat3(1, 1), 4);
VERIFY_IS_EQUAL(mat3(1, 2), 5);
VERIFY_IS_EQUAL(mat4(0,0), 0);
VERIFY_IS_EQUAL(mat4(0,1), 1);
VERIFY_IS_EQUAL(mat4(0,2), 2);
VERIFY_IS_EQUAL(mat4(1,0), 3);
VERIFY_IS_EQUAL(mat4(1,1), 4);
VERIFY_IS_EQUAL(mat4(1,2), 5);
VERIFY_IS_EQUAL(mat4(0, 0), 0);
VERIFY_IS_EQUAL(mat4(0, 1), 1);
VERIFY_IS_EQUAL(mat4(0, 2), 2);
VERIFY_IS_EQUAL(mat4(1, 0), 3);
VERIFY_IS_EQUAL(mat4(1, 1), 4);
VERIFY_IS_EQUAL(mat4(1, 2), 5);
mat1.setZero();
mat2.setZero();
mat1 = mat3;
mat2 = mat4;
VERIFY_IS_EQUAL(mat1(0,0), 0);
VERIFY_IS_EQUAL(mat1(0,1), 1);
VERIFY_IS_EQUAL(mat1(0,2), 2);
VERIFY_IS_EQUAL(mat1(1,0), 3);
VERIFY_IS_EQUAL(mat1(1,1), 4);
VERIFY_IS_EQUAL(mat1(1,2), 5);
VERIFY_IS_EQUAL(mat1(0, 0), 0);
VERIFY_IS_EQUAL(mat1(0, 1), 1);
VERIFY_IS_EQUAL(mat1(0, 2), 2);
VERIFY_IS_EQUAL(mat1(1, 0), 3);
VERIFY_IS_EQUAL(mat1(1, 1), 4);
VERIFY_IS_EQUAL(mat1(1, 2), 5);
VERIFY_IS_EQUAL(mat2(0,0), 0);
VERIFY_IS_EQUAL(mat2(0,1), 1);
VERIFY_IS_EQUAL(mat2(0,2), 2);
VERIFY_IS_EQUAL(mat2(1,0), 3);
VERIFY_IS_EQUAL(mat2(1,1), 4);
VERIFY_IS_EQUAL(mat2(1,2), 5);
VERIFY_IS_EQUAL(mat2(0, 0), 0);
VERIFY_IS_EQUAL(mat2(0, 1), 1);
VERIFY_IS_EQUAL(mat2(0, 2), 2);
VERIFY_IS_EQUAL(mat2(1, 0), 3);
VERIFY_IS_EQUAL(mat2(1, 1), 4);
VERIFY_IS_EQUAL(mat2(1, 2), 5);
}
static void test_3d()
{
Tensor<int, 3> mat1(2,3,7);
Tensor<int, 3, RowMajor> mat2(2,3,7);
static void test_3d() {
Tensor<int, 3> mat1(2, 3, 7);
Tensor<int, 3, RowMajor> mat2(2, 3, 7);
int val = 0;
for (int i = 0; i < 2; ++i) {
for (int j = 0; j < 3; ++j) {
for (int k = 0; k < 7; ++k) {
mat1(i,j,k) = val;
mat2(i,j,k) = val;
mat1(i, j, k) = val;
mat2(i, j, k) = val;
val++;
}
}
}
int col_major[2*3*7] = {0};
int row_major[2*3*7] = {0};
int col_major[2 * 3 * 7] = {0};
int row_major[2 * 3 * 7] = {0};
TensorMap<Tensor<int, 3> > mat3(col_major, 2, 3, 7);
TensorMap<Tensor<int, 3, RowMajor> > mat4(row_major, 2, 3, 7);
@@ -156,8 +159,8 @@ static void test_3d()
for (int i = 0; i < 2; ++i) {
for (int j = 0; j < 3; ++j) {
for (int k = 0; k < 7; ++k) {
VERIFY_IS_EQUAL(mat3(i,j,k), val);
VERIFY_IS_EQUAL(mat4(i,j,k), val);
VERIFY_IS_EQUAL(mat3(i, j, k), val);
VERIFY_IS_EQUAL(mat4(i, j, k), val);
val++;
}
}
@@ -172,16 +175,15 @@ static void test_3d()
for (int i = 0; i < 2; ++i) {
for (int j = 0; j < 3; ++j) {
for (int k = 0; k < 7; ++k) {
VERIFY_IS_EQUAL(mat1(i,j,k), val);
VERIFY_IS_EQUAL(mat2(i,j,k), val);
VERIFY_IS_EQUAL(mat1(i, j, k), val);
VERIFY_IS_EQUAL(mat2(i, j, k), val);
val++;
}
}
}
}
static void test_same_type()
{
static void test_same_type() {
Tensor<int, 1> orig_tensor(5);
Tensor<int, 1> dest_tensor(5);
orig_tensor.setRandom();
@@ -218,12 +220,11 @@ static void test_same_type()
VERIFY_IS_EQUAL(orig_map.data(), orig_data);
VERIFY_IS_EQUAL(dest_map.data(), dest_data);
for (int i = 0; i < 5; ++i) {
VERIFY_IS_EQUAL(dest[i], i+1);
VERIFY_IS_EQUAL(dest[i], i + 1);
}
}
static void test_auto_resize()
{
static void test_auto_resize() {
Tensor<int, 1> tensor1;
Tensor<int, 1> tensor2(3);
Tensor<int, 1> tensor3(5);
@@ -246,9 +247,7 @@ static void test_auto_resize()
}
}
static void test_compound_assign()
{
static void test_compound_assign() {
Tensor<int, 1> start_tensor(10);
Tensor<int, 1> offset_tensor(10);
start_tensor.setRandom();
@@ -321,9 +320,8 @@ static void test_std_initializers_tensor() {
VERIFY_IS_EQUAL(b(1, 2), 5);
Eigen::Tensor<int, 3> c(3, 2, 4);
c.setValues({{{0, 1, 2, 3}, {4, 5, 6, 7}},
{{10, 11, 12, 13}, {14, 15, 16, 17}},
{{20, 21, 22, 23}, {24, 25, 26, 27}}});
c.setValues(
{{{0, 1, 2, 3}, {4, 5, 6, 7}}, {{10, 11, 12, 13}, {14, 15, 16, 17}}, {{20, 21, 22, 23}, {24, 25, 26, 27}}});
VERIFY_IS_EQUAL(c(0, 0, 0), 0);
VERIFY_IS_EQUAL(c(0, 0, 1), 1);
VERIFY_IS_EQUAL(c(0, 0, 2), 2);
@@ -350,8 +348,7 @@ static void test_std_initializers_tensor() {
VERIFY_IS_EQUAL(c(2, 1, 3), 27);
}
EIGEN_DECLARE_TEST(cxx11_tensor_assign)
{
EIGEN_DECLARE_TEST(cxx11_tensor_assign) {
CALL_SUBTEST(test_1d());
CALL_SUBTEST(test_2d());
CALL_SUBTEST(test_3d());