diff options
| author |  Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-06-03 17:28:33 -0700 |
|---|---|---|
| committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-06-03 17:28:33 -0700 |
| commit | 36a45008223ee9d086b9a4e8b885e8882ccf319d (patch) | |
| tree | 94c60b39a948ede020e63d5e3ca9648d84614293 | |
| parent | c2a102345f627e4cd1908dad03e6ef0cbb2170c0 (diff) | |
| parent | dc03b8f3a16571a6e776d44529d3e69851db7f97 (diff) | |
Merged in ibab/eigen (pull request PR-192)
Add generic scan method
| -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> |