diff options
| -rw-r--r-- | arm_compute/core/TensorShape.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/arm_compute/core/TensorShape.h b/arm_compute/core/TensorShape.h index 0340e1a644..d531dcf462 100644 --- a/arm_compute/core/TensorShape.h +++ b/arm_compute/core/TensorShape.h @@ -160,7 +160,7 @@ public: TensorShape collapsed_from(size_t start) const { TensorShape copy(*this); - copy.collapse(num_dimensions(), start); + copy.collapse(num_dimensions() - start, start); return copy; } |