diff options
| author |  Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-01-13 14:24:37 -0800 |
|---|---|---|
| committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-01-13 14:24:37 -0800 |
| commit | 9f013a9d86ad5cf82939bfeab2223652a821c448 (patch) | |
| tree | 47e8f1d08fe1e1cf79cb793e43937592782e8c9d /unsupported/Eigen/CXX11/src/Tensor/TensorArgMax.h | |
| parent | 79b69b7444cfae2f7631e873e822cdca6f4e355f (diff) | |
Properly record the rank of reduced tensors in the tensor traits.
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorArgMax.h')
| -rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorArgMax.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorArgMax.h b/unsupported/Eigen/CXX11/src/Tensor/TensorArgMax.h index c783aab97..781a37e34 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorArgMax.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorArgMax.h @@ -134,7 +134,7 @@ struct traits<TensorTupleReducerOp<ReduceOp, Dims, XprType> > : public traits<Xp typedef Index Scalar; typedef typename XprType::Nested Nested; typedef typename remove_reference<Nested>::type _Nested; - static const int NumDimensions = XprTraits::NumDimensions; + static const int NumDimensions = XprTraits::NumDimensions - array_size<Dims>::value; static const int Layout = XprTraits::Layout; }; |