Fix the problem with unflatten numpy arrays (#695)

016780f4 · tsocha · Scott Cyphers · 1447b578 · 016780f4
Commit 016780f4 authored Mar 20, 2018 by tsocha Committed by Scott Cyphers Mar 20, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

runtime.py python/ngraph/runtime.py +1 -1

No files found.
--- a/python/ngraph/runtime.py
+++ b/python/ngraph/runtime.py
@@ -114,7 +114,7 @@ class Computation:

        buffer_size = Computation._get_buffer_size(
            tensor_view.element_type, tensor_view.element_count)
-        tensor_view.write(util.numpy_to_c(value), 0, buffer_size)
+        tensor_view.write(util.numpy_to_c(np.ascontiguousarray(value)), 0, buffer_size)

    @staticmethod
    def _read_tensor_view_to_ndarray(tensor_view, output):