mirror/eigen
2
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2024-12-15 07:10:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add generic scan method

Browse Source
This commit is contained in:
Igor Babuschkin 2016-06-03 17:37:04 +01:00
parent 8d97ba6b22
commit dc03b8f3a1
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
unsupported/Eigen/CXX11/src/Tensor/TensorBase.h
View File

@ -498,6 +498,13 @@ class TensorBase<Derived, ReadOnlyAccessors>
return TensorScanProdOp(derived(), axis); 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. // Reductions.
template <typename Dims> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE template <typename Dims> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
const TensorReductionOp<internal::SumReducer<CoeffReturnType>, const Dims, const Derived> const TensorReductionOp<internal::SumReducer<CoeffReturnType>, const Dims, const Derived>