You can probably transpose the array to make the target axis the innermost
one, and then reduce over it. Perhaps a specialized reduction will be
faster, but this will be a good first approximation.
On Fri, Aug 21, 2015 at 5:30 AM, Joseph Austerweil <
First, thank you so much for your hard work on the package.
It is a fantastic resource.
A bit of context:
My lab and I are working on building approximations
for Bayesian nonparametric models implemented in
Python using pyopencl (as far as I can tell, none exist).
So far our package is in its infancy,
but hope it will be useful for others in about a year.
One operation that arises frequently is reducing over
a single axis of a multi-dimensional array
(for example, lets say we have the log probability
calculated for each element of a N x D observation matrix
and want to sum over D to get the log probability of each object
-- more frequently I have 3-d arrays
and am summing over one axis of it).
Is it possible to use clarray's reduce sum
capabilities to sum over a single axis?
Currently, I wrote my own for a simple 3-d case,
but it's not nearly as robust as the one provided in
Thank you in advance,
Assistant Professor of Cognitive, Linguistic, and Psychological Sciences
PyOpenCL mailing list