Python: Multiplicando uma lista de vetores por uma lista de matrizes como uma única operação de matriz
Eu tenho uma lista de 100N-dimensional
vetores e uma lista de100 MxN
matrizes. Então você pode pensar nas duas estruturas de dados como um100xN
lista (ou numpy array) e um100xMxN
lista (ou matriz numpy).
O que eu quero fazer é pegar o produto escalar de cada vetor e sua matriz correspondente, de modo que a saída deva ser100 M-dimensional
matrizes (ou seja, uma100xM
lista ou matriz numpy).
No entanto, não tenho muita certeza de como fazer isso. Não quero fazer iterativamente, por razões óbvias sobre eficiência. Eu também sei que não é multiplicação de matrizes básica. Eu acho que posso querer usarnp.einsum
, mas não estou muito familiarizado com isso.
Alguém quer ajudar?