ENH: writeable and owndata flags

Author: ev-br

torch_np/_ndarray.py

@@ -115,12 +115,18 @@ def flags(self):
 
         # check if F contiguous
         from itertools import accumulate
-        f_strides = tuple(accumulate(list(self._tensor.shape), func=lambda x, y: x*y))
+
+        f_strides = tuple(accumulate(list(self._tensor.shape), func=lambda x, y: x * y))
         f_strides = (1,) + f_strides[:-1]
         is_f_contiguous = f_strides == self._tensor.stride()
 
-        return Flags({"C_CONTIGUOUS": self._tensor.is_contiguous(),
-                      "F_CONTIGUOUS": is_f_contiguous,}
+        return Flags(
+            {
+                "C_CONTIGUOUS": self._tensor.is_contiguous(),
+                "F_CONTIGUOUS": is_f_contiguous,
+                "OWNDATA": self._tensor._base is None,
+                "WRITEABLE": True,  # pytorch does not have readonly tensors
+            }
         )
 
     @property