vectorizing functions with multiple return values?
Nobody has claimed this yet.
- Dominant language
- C++
- Stars
- 18k
- Forks
- 2.3k
- Avg merge
- 5d 17h
- Merged PRs (30d)
- 10
Description
My goal is to provide a vectorized Python binding to C++ functions that return more than a single value, like
void ang2vec (double theta, double phi, double &x, double &y, double &z)
{
x=sin(theta)*cos(phi);
y=sin(theta)*sin(phi);
z=cos(theta);
}
py::vectorize(ang2vec) does not work, producing error messages like
"pybind11/numpy.h:732:5: error: forming pointer to reference type ‘double&’".
I tried to translate this into a more Python-like interface:
tuple<double,double,double> ang2vec2 (double theta, double phi)
{
return tuple<double,double,double>(sin(theta)*cos(phi),sin(theta)*sin(phi),cos(theta));
}
but unfortunately vectorizing this fails as well.
Are there fundamental obstacles to supporting this scenario, or am I just the first one trying to do this? :)
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start by tracing py::vectorize and the failure reported at pybind11/numpy.h:732 for both the reference-argument and tuple-return examples. Determine whether multiple outputs can be supported and identify the tests needed to establish the expected Python-facing behavior; done means a documented, tested resolution of this use case.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- cpp, python
- Domain
- api
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100