Numpy, python: espande automaticamente le dimensioni degli array durante la trasmissione

Question

Considerare il seguente esercizio nella trasmissione di array Numpy.

import numpy as np 
v = np.array([[1.0, 2.0]]).T # column array 

A2 = np.random.randn(2,10) # 2D array 
A3 = np.random.randn(2,10,10) # 3D 

v * A2 # works great 

# causes error: 
v * A3 # error 

Conosco le regole NumPy per la radiodiffusione, e ho familiarità con bsxfun funzionalità in Matlab. Capisco perché il tentativo di trasmettere un (2,1) array in un array (2, N, N) fallisce e che devo rimodellare l'array (2,1) in un array (2,1,1) prima di questo la trasmissione passa attraverso

La mia domanda è: c'è un modo per dire a Python di eseguire il rilievo automatico della dimensionalità di un array quando tenta di trasmettere, senza che io debba specificarlo specificatamente?

ho non voglio esplicitamente paio il (2,1) di vettore con l'array multidimensionale sta andando in onda contro --- altrimenti avrei potuto fare qualcosa di stupido e brutto come assurdamente mult_v_A = lambda v,A: v.reshape([v.size] + [1]*(A.ndim-1)) * A. Non so in anticipo se la matrice "A" sarà 2D o 3D o N-D.

La funzionalità di trasmissione di Matlab bsxfun include implicitamente le dimensioni necessarie, quindi spero che ci sia qualcosa che potrei fare in Python.

Answer 1

E 'brutto, ma questo funzionerà :

(v.T * A3.T).T 

Se non si dà alcun argomento, recepisce inverte la tupla forma, in modo da poter ora avvalersi delle norme di trasmissione di fare la loro magia. L'ultima trasposizione restituisce tutto nell'ordine corretto.

Answer 2

La trasmissione NumPy aggiunge ulteriori assi a sinistra.

Quindi, se ad organizzare gli array in modo che le assi comuni sono a destra e gli assi broadcastable sono a sinistra, quindi è possibile utilizzare la trasmissione senza alcun problema:

import numpy as np 
v = np.array([[1.0, 2.0]]) # shape (1, 2) 

A2 = np.random.randn(10,2) # shape (10, 2) 
A3 = np.random.randn(10,10,2) # shape (10, 10, 2) 

v * A2 # shape (10, 2) 

v * A3 # shape (10, 10, 2)