cxx11_tensor_layout_swap.cpp File Reference

#include "main.h"
#include <Eigen/CXX11/Tensor>

Functions
static void	test_simple_swap ()
static void	test_swap_as_lvalue ()
	EIGEN_DECLARE_TEST (cxx11_tensor_layout_swap)

Function Documentation

EIGEN_DECLARE_TEST()

EIGEN_DECLARE_TEST

(

cxx11_tensor_layout_swap

)

                                             {
  CALL_SUBTEST(test_simple_swap());
  CALL_SUBTEST(test_swap_as_lvalue());
}

References CALL_SUBTEST, test_simple_swap(), and test_swap_as_lvalue().

test_simple_swap()

static void test_simple_swap ( )

static

                               {
  Tensor<float, 3, ColMajor> tensor(2, 3, 7);
  tensor.setRandom();
 
  Tensor<float, 3, RowMajor> tensor2 = tensor.swap_layout();
  VERIFY_IS_EQUAL(tensor.dimension(0), tensor2.dimension(2));
  VERIFY_IS_EQUAL(tensor.dimension(1), tensor2.dimension(1));
  VERIFY_IS_EQUAL(tensor.dimension(2), tensor2.dimension(0));
 
  for (int i = 0; i < 2; ++i) {
    for (int j = 0; j < 3; ++j) {
      for (int k = 0; k < 7; ++k) {
        VERIFY_IS_EQUAL(tensor(i, j, k), tensor2(k, j, i));
      }
    }
  }
}

References Eigen::Tensor< Scalar_, NumIndices_, Options_, IndexType_ >::dimension(), i, j, k, Eigen::TensorBase< Derived, AccessLevel >::setRandom(), Eigen::TensorBase< Derived, AccessLevel >::swap_layout(), and VERIFY_IS_EQUAL.

Referenced by EIGEN_DECLARE_TEST().

test_swap_as_lvalue()

static void test_swap_as_lvalue ( )

static

                                  {
  Tensor<float, 3, ColMajor> tensor(2, 3, 7);
  tensor.setRandom();
 
  Tensor<float, 3, RowMajor> tensor2(7, 3, 2);
  tensor2.swap_layout() = tensor;
  VERIFY_IS_EQUAL(tensor.dimension(0), tensor2.dimension(2));
  VERIFY_IS_EQUAL(tensor.dimension(1), tensor2.dimension(1));
  VERIFY_IS_EQUAL(tensor.dimension(2), tensor2.dimension(0));
 
  for (int i = 0; i < 2; ++i) {
    for (int j = 0; j < 3; ++j) {
      for (int k = 0; k < 7; ++k) {
        VERIFY_IS_EQUAL(tensor(i, j, k), tensor2(k, j, i));
      }
    }
  }
}

References Eigen::Tensor< Scalar_, NumIndices_, Options_, IndexType_ >::dimension(), i, j, k, Eigen::TensorBase< Derived, AccessLevel >::setRandom(), Eigen::TensorBase< Derived, AccessLevel >::swap_layout(), and VERIFY_IS_EQUAL.

Referenced by EIGEN_DECLARE_TEST().