GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-06-04 18:54:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added sanity check

Browse Source
This commit is contained in:
Benoit Steiner 2015-11-13 09:07:27 -08:00
parent 1e1755352d
commit f1fbd74db9
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
unsupported/Eigen/CXX11/src/Tensor/TensorReverse.h
View File

@ -66,7 +66,7 @@ class TensorReverseOp : public TensorBase<TensorReverseOp<ReverseDimensions,
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorReverseOp( EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorReverseOp(
const XprType& expr, const ReverseDimensions& reverse_dims) const XprType& expr, const ReverseDimensions& reverse_dims)
: m_xpr(expr), m_reverse_dims(reverse_dims) {} : m_xpr(expr), m_reverse_dims(reverse_dims) { }
EIGEN_DEVICE_FUNC EIGEN_DEVICE_FUNC
const ReverseDimensions& reverse() const { return m_reverse_dims; } const ReverseDimensions& reverse() const { return m_reverse_dims; }
@ -119,6 +119,9 @@ struct TensorEvaluator<const TensorReverseOp<ReverseDimensions, ArgType>, Device
const Device& device) const Device& device)
: m_impl(op.expression(), device), m_reverse(op.reverse()) : m_impl(op.expression(), device), m_reverse(op.reverse())
{ {
// Reversing a scalar isn't supported yet. It would be a no-op anyway.
EIGEN_STATIC_ASSERT(NumDims > 0, YOU_MADE_A_PROGRAMMING_MISTAKE);
// Compute strides // Compute strides
m_dimensions = m_impl.dimensions(); m_dimensions = m_impl.dimensions();
if (static_cast<int>(Layout) == static_cast<int>(ColMajor)) { if (static_cast<int>(Layout) == static_cast<int>(ColMajor)) {