On 30 January 2018 at 17:22, Andrew Nelson wrote:
> Hi all,
> I have a matrix multiplication that I'd like to optimize.
>
> I have a matrix `a` (dtype=complex) with shape (N, M, 2, 2). I'd like to
> do the following multiplication:
>
> a[:, 0] @ a[:, 1] @ ... @ a[:, M-1]
>
> where the first dimen
Hi all,
I have a matrix multiplication that I'd like to optimize.
I have a matrix `a` (dtype=complex) with shape (N, M, 2, 2). I'd like to do
the following multiplication:
a[:, 0] @ a[:, 1] @ ... @ a[:, M-1]
where the first dimension, N, is element wise (and hopefully vectorisable)
and M>=2. So
