yeah i can't find the post i'm looking for. oh well, here's the reply anyway:

well i tried learning the matrix being dotted but it seems like there's something fundamentally wrong with my idea. (redundancy? autograd bug?)

not really in the mood to debug this, especially since i realized my initial idea wouldn't work after i had started coding it.