40a41,57
>
>
> // Internal optimization test for whether the input is trivially broadcastable:
> py::enum_<py::detail::broadcast_trivial>(m, "trivial")
> .value("f_trivial", py::detail::broadcast_trivial::f_trivial)
> .value("c_trivial", py::detail::broadcast_trivial::c_trivial)
> .value("non_trivial", py::detail::broadcast_trivial::non_trivial);
> m.def("vectorized_is_trivial", [](
> py::array_t<int, py::array::forcecast> arg1,
> py::array_t<float, py::array::forcecast> arg2,
> py::array_t<double, py::array::forcecast> arg3
> ) {
> size_t ndim;
> std::vector<size_t> shape;
> std::array<py::buffer_info, 3> buffers {{ arg1.request(), arg2.request(), arg3.request() }};
> return py::detail::broadcast(buffers, ndim, shape);
> });