[Numpy-discussion] vectorization of vectorization
John Salvatier
jsalvati@u.washington....
Thu Aug 19 16:32:42 CDT 2010
This seems simple, so perhaps I am missing something, but what about this:
special.iv(orders[:, newaxis], arange(1e6)[newaxis, : ], powers[:,
newaxis]).prod(axis = 2)
This will probably use up a bit slow/memoryintensive, so you probably want
to use numexpr to speed it up a bit.
On Thu, Aug 19, 2010 at 2:22 PM, sm lkd <thusweare@hotmail.com> wrote:
> Hello,
>
> Here's my problem: for each value t of an array (from 0 to 1e6) a smaller
> array is computed (size between 2-6). To compute the smaller array, I have a
> function (which can be easily vectorized if necessary) which takes t and an
> array of powers of t. The return is an array of modified Bessel function
> values, i.e.:
>
> def funct(order, t, power):
> return special.iv(order, t)**power
>
> Note that order and power are arrays after this vectorization:
> vec_func = sp.vectorization(func)
>
> Right this is how it's used:
> for i in range(1000000):
> y[i] = vec_func(orders, t, powers).prod()
>
> Incredibly slow.
>
> Of course, it is desirable to vectorize it it terms of t. I have tried
> different methods but still cannot make it work. Any suggestions or ointers?
>
> Thank you.
>
> _______________________________________________
> NumPy-Discussion mailing list
> NumPy-Discussion@scipy.org
> http://mail.scipy.org/mailman/listinfo/numpy-discussion
>
>
-------------- next part --------------
An HTML attachment was scrubbed...
URL: http://mail.scipy.org/pipermail/numpy-discussion/attachments/20100819/833e8ea5/attachment-0001.html
More information about the NumPy-Discussion
mailing list