Method numpy

imperative/python/src/tensor.cpp:696–719

}

PyObject* TensorWrapper::numpy() {
    auto hv = m_tensor->numpy();
    if (!hv) {
        if (TransformationManager::get_instance()
                    .segments[TransformationManager::Segment::Eval]
                    .size() > 1) {
            PyErr_SetString(
                    PyExc_ValueError,
                    "tensor invalid, can not infer value of this tensor under "
                    "trace(symbolic=True). You can try to use trace(symbolic=False) to "
                    "avoid this issue.");
        } else {
            PyErr_SetString(PyExc_ValueError, "tensor invalid");
        }
        return nullptr;
    }
    auto arr = py::reinterpret_steal<py::array>(
            npy::ndarray_from_tensor(hv->as_nd(true), npy::ShareType::TRY_SHARE));
    if (hv->shape().is_scalar()) {
        mgb_assert(PyArray_Check(arr.ptr()));
        return PyArray_Squeeze(reinterpret_cast<PyArrayObject*>(arr.ptr()));
    }
    return arr.release().ptr();
}

void TensorWrapper::reset(PyObject* tensor) {
    TensorWrapper* t = TensorWrapper::try_cast(tensor);