From 78f37ca03ce9a6e348858d1b7d7b2c2b0e7fec93 Mon Sep 17 00:00:00 2001 From: Igor Babuschkin Date: Fri, 1 Jul 2016 17:34:31 +0100 Subject: [PATCH] Expose real and imag methods on Tensors --- unsupported/Eigen/CXX11/src/Tensor/TensorBase.h | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h b/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h index 73bfac40e..142778df8 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h @@ -209,6 +209,18 @@ class TensorBase return unaryExpr(internal::bind2nd_op >(exponent)); } + EIGEN_DEVICE_FUNC + EIGEN_STRONG_INLINE const TensorCwiseUnaryOp, const Derived> + real() const { + return unaryExpr(internal::scalar_real_op()); + } + + EIGEN_DEVICE_FUNC + EIGEN_STRONG_INLINE const TensorCwiseUnaryOp, const Derived> + imag() const { + return unaryExpr(internal::scalar_imag_op()); + } + EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp >, const Derived> operator+ (Scalar rhs) const {