diff options
Diffstat (limited to 'ndarray.c')
-rw-r--r-- | ndarray.c | 4 |
1 files changed, 4 insertions, 0 deletions
@@ -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; |