hackswip

1 files changed, 4 insertions, 0 deletions

diff --git a/ndarray.c b/ndarray.c
index dee985c..614e3fe 100644
--- a/ndarray.c
+++ b/ndarray.c
@@ -78,6 +78,10 @@ int mg2di_ndarray_alloc(NDArray **result, unsigned int dims,
     if (flags & NDARRAY_ALLOC_ZERO)
         memset(a->priv->data, 0, a->priv->alloc_size);
 
+    for (int i = 0; i < dims; i++)
+        if (size[i] == 1)
+            a->priv->stride[i] = 0;
+
     a->dims   = dims;
 
     a->data   = a->priv->data;