diff options
author | Rasmus Munk Larsen <rmlarsen@google.com> | 2018-05-14 16:07:13 -0700 |
---|---|---|
committer | Rasmus Munk Larsen <rmlarsen@google.com> | 2018-05-14 16:07:13 -0700 |
commit | b8c8e5f436743ac6a6a5ed0ad7ec5cce7dd00248 (patch) | |
tree | 8dd48fd88fa87fd9c492bfcbf5fd3ed4ea8cb150 /unsupported/Eigen/CXX11/src/Tensor/TensorBase.h | |
parent | 6118c6ff4f898dad99564ffdeb99036beb2ff0ea (diff) |
Add vectorized clip functor for Eigen Tensors.
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorBase.h')
-rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorBase.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h b/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h index 1d459a3a0..c8dc3bbc5 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h @@ -210,6 +210,12 @@ class TensorBase<Derived, ReadOnlyAccessors> } EIGEN_DEVICE_FUNC + EIGEN_STRONG_INLINE const TensorCwiseUnaryOp<internal::scalar_clip_op<Scalar>, const Derived> + clip(Scalar min, Scalar max) const { + return unaryExpr(internal::scalar_clip_op<Scalar>(min, max)); + } + + EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp<internal::scalar_conjugate_op<Scalar>, const Derived> conjugate() const { return unaryExpr(internal::scalar_conjugate_op<Scalar>()); |