ENH: Make einsum return a writeable view whenever the input is writea…

…ble.
aman-thakral · Jan 1, 2015 · 59cab3e · 59cab3e
1 parent 061feab
commit 59cab3e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/numpy/core/src/multiarray/einsum.c.src b/numpy/core/src/multiarray/einsum.c.src
@@ -2158,7 +2158,8 @@ get_single_op_view(PyArrayObject *op, int  iop, char *labels,
                                 PyArray_DESCR(op),
                                 ndim_output, new_dims, new_strides,
                                 PyArray_DATA(op),
-                                0, (PyObject *)op);
+                                PyArray_ISWRITEABLE(op) ? NPY_ARRAY_WRITEABLE : 0,
+                                (PyObject *)op);
 
         if (*ret == NULL) {
             return 0;