I had this gross reshape/tensor product/transpose stuff on huge matrices, and I knew it was making
intermediate copies of the matrices that I didn't want to. So I tried out np.einsum
, and
I think it actually turned out simpler than thinking through the other matrix manipulation.
Here are some quick notes.