diff --git a/mindspore/ccsrc/pybind_api/ir/tensor_py.cc b/mindspore/ccsrc/pybind_api/ir/tensor_py.cc index 2e68f24fb9..3e9707c41c 100644 --- a/mindspore/ccsrc/pybind_api/ir/tensor_py.cc +++ b/mindspore/ccsrc/pybind_api/ir/tensor_py.cc @@ -137,7 +137,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; } @@ -162,6 +162,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_; };