diff --git a/mindspore/ccsrc/pybind_api/ir/tensor_py.cc b/mindspore/ccsrc/pybind_api/ir/tensor_py.cc index 2eeba606bf..13ba6314b5 100644 --- a/mindspore/ccsrc/pybind_api/ir/tensor_py.cc +++ b/mindspore/ccsrc/pybind_api/ir/tensor_py.cc @@ -138,7 +138,7 @@ class TensorDataNumpy : public TensorData { ssize_t ndim() const override { return buffer_.ndim; } /// Data pointer. - void *data() override { return buffer_.ptr; } + void *data() override { return buffer_data(); } const void *const_data() const override { return buffer_.ptr; } @@ -163,6 +163,9 @@ class TensorDataNumpy : public TensorData { return py::array(py::dtype(buffer_), buffer_.shape, buffer_.strides, buffer_.ptr, dummyOwner); } + private: + void *buffer_data() { return buffer_.ptr; } + // The internal buffer. py::buffer_info buffer_; };