diff options
author | Igor Babuschkin <igor@babuschk.in> | 2016-06-03 17:37:04 +0100 |
---|---|---|
committer | Igor Babuschkin <igor@babuschk.in> | 2016-06-03 17:37:04 +0100 |
commit | dc03b8f3a16571a6e776d44529d3e69851db7f97 (patch) | |
tree | a0370300e668b74cdd90f1712ad7df6be0b5a894 /unsupported | |
parent | 8d97ba6b2251aabf325ff74f24959ceaa85cf11e (diff) |
Add generic scan method
Diffstat (limited to 'unsupported')
-rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorBase.h | 7 |
1 files changed, 7 insertions, 0 deletions
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<Derived, ReadOnlyAccessors> return TensorScanProdOp(derived(), axis); } + template <typename Reducer> + EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE + const TensorScanOp<Reducer, const Derived> + scan(const Index& axis, const Reducer& reducer) const { + return TensorScanOp<Reducer, const Derived>(derived(), axis, reducer); + } + // Reductions. template <typename Dims> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorReductionOp<internal::SumReducer<CoeffReturnType>, const Dims, const Derived> |