devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Return the proper size (ie 1) for tensors of rank 0

Browse Source
This commit is contained in:
Benoit Steiner
2016-04-29 18:14:33 -07:00
parent d14105f158
commit f100d1494c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
unsupported/Eigen/CXX11/src/Tensor/TensorDimensions.h
View File

@@ -275,7 +275,7 @@ struct DSizes : array<DenseIndex, NumDims> {
} }
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE DenseIndex TotalSize() const { EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE DenseIndex TotalSize() const {
return internal::array_prod(*static_cast<const Base*>(this)); return (NumDims == 0) ? 1 : internal::array_prod(*static_cast<const Base*>(this));
} }
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE DSizes() { EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE DSizes() {