Numpy, python: yayın sırasında dizilerin boyutlarını otomatik olarak genişlet

Question

Numpy dizisi yayınında aşağıdaki alıştırmaları göz önünde bulundurun.

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 

Ben yayın için Numpy kurallarını biliyorum ve Matlab bsxfun işlevselliği ile tanıdık. Bir (2, N, N) dizisine bir (2,1) dizisini göndermeyi denemenin neden başarısız olduğunu ve (2,1) dizisini bundan önce bir (2,1,1) diziye yeniden eklemem gerektiğini anlıyorum. yayın geçer.

Soruma şudur: Python'u yayınlamaya çalışırken dizinin boyutlarını otomatik olarak doldurması için herhangi bir yol var mı?

aksi takdirde ben aptal ve mult_v_A = lambda v,A: v.reshape([v.size] + [1]*(A.ndim-1)) * A gibi saçma çirkin bir şey yapabileceğini --- o karşı yayınlanacak olacak çok boyutlu dizi (2,1) vektörü için açıkça çift istemiyoruz . "A" dizisi 2D veya 3D veya N-D olacaksa, zamanın önünde bilmiyorum. Matlab'ın bsxfun yayın işlevselliği, boyutları gerektiği gibi zımbalı bir şekilde pedler, bu yüzden Python'da yapabileceğim bir şey olduğunu umuyorum. Numpe yayıncılık, sol tarafta ek eksenler ekler.

Answer 1

Bu çirkin, ama bu çalışacaktır : bunu herhangi bir bağımsız değişkeni vermezseniz artık onların büyü yapmak için yayın kurallarına güvenebilirsiniz böylece

(v.T * A3.T).T 

, aktarılmasını, şekil tuple tersine çevirir. Son aktarım her şeyi doğru sıraya döndürür.

Answer 2

Paylaşılan eksenleri sağda ve broadcastable eksenleri solda böylece diziler düzenlemek eğer

Yani, o zaman hiçbir sorun ile yayın kullanabilirsiniz:

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)