Function ZeroDimArrayToScalar

tensorflow/python/lib/core/py_seq_tensor.cc:71–79

If the input is a zero dimensional PyArray return it converted to a scalar. Otherwise return the input and increment its reference count. Users must Py_DECREF the output of this method.

// Otherwise return the input and increment its reference count.
// Users must Py_DECREF the output of this method.
PyObject* ZeroDimArrayToScalar(PyObject* obj) {
  if (PyArray_IsZeroDim(obj) && !PyArray_IsScalar(obj, Generic)) {
    auto pyarray_obj = reinterpret_cast<PyArrayObject*>(obj);
    obj = PyArray_ToScalar(PyArray_DATA(pyarray_obj), pyarray_obj);
  } else {
    Py_INCREF(obj);
  }
  return obj;
}

// Converts Python object `c` that should hold a Python string into a
// C++ string in *out.  Returns nullptr on success, or a message on error.