diff --git a/arm_compute/core/utils/misc/ShapeCalculator.h b/arm_compute/core/utils/misc/ShapeCalculator.h
index adf5309..4756ff4 100644
--- a/arm_compute/core/utils/misc/ShapeCalculator.h
+++ b/arm_compute/core/utils/misc/ShapeCalculator.h
@@ -803,6 +803,13 @@
     return *data;
 }
 
+inline TensorShape calculate_unstack_shape(TensorShape input_shape, unsigned int axis)
+{
+    ARM_COMPUTE_ERROR_ON(axis > input_shape.num_dimensions());
+    input_shape.remove_dimension(axis);
+    return input_shape;
+}
+
 template <typename T>
 inline TensorShape calculate_depth_concatenate_shape(const std::vector<T *> &inputs_vector)
 {