GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-08-12 11:49:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix a bug in TensorGenerator for 1d tensors

Browse Source
This commit is contained in:
Eugene Zhulenev 2019-03-11 09:42:01 -07:00
parent 0f8bfff23d
commit 899c16fa2c
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
unsupported/Eigen/CXX11/src/Tensor/TensorGenerator.h
View File

@ -208,6 +208,9 @@ struct TensorEvaluator<const TensorGeneratorOp<Generator, ArgType>, Device>
coords[is_col_major ? 0 : NumDims - 1] = coords[is_col_major ? 0 : NumDims - 1] =
initial_coords[is_col_major ? 0 : NumDims - 1]; initial_coords[is_col_major ? 0 : NumDims - 1];
// For the 1d tensor we need to generate only one inner-most dimension.
if (NumDims == 1) break;
// Update offset. // Update offset.
for (Index i = 1; i < NumDims; ++i) { for (Index i = 1; i < NumDims; ++i) {
if (++it[i].count < it[i].size) { if (++it[i].count < it[i].size) {