From dc03b8f3a16571a6e776d44529d3e69851db7f97 Mon Sep 17 00:00:00 2001 From: Igor Babuschkin Date: Fri, 3 Jun 2016 17:37:04 +0100 Subject: Add generic scan method --- unsupported/Eigen/CXX11/src/Tensor/TensorBase.h | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorBase.h') diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h b/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h index 12f8a1499..eafd6f6f1 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h @@ -498,6 +498,13 @@ class TensorBase return TensorScanProdOp(derived(), axis); } + template + EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE + const TensorScanOp + scan(const Index& axis, const Reducer& reducer) const { + return TensorScanOp(derived(), axis, reducer); + } + // Reductions. template EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorReductionOp, const Dims, const Derived> -- cgit v1.2.3