Разделение многообразного многомерного массива различной формы

Question

Я попробовал второй пример ниже, и это не сработало, но пример небольшого масштаба для деления массива numpy с другим отлично работает. Какая разница? Как я могу заставить фактическое подразделение работать?

Небольшой масштаб пример того, что я пытаюсь сделать:

>>> import numpy as np 
>>> a = np.asarray([[1, 2, 3], [4, 5, 6]]) 
>>> b = np.asarray([2, 2, 2]) 
>>> a/b   # works as expected 
array([[0, 1, 1], 
     [2, 2, 3]]) 

Фактическое разделение, что я пытаюсь получить работу

>>> a = np.random.random((9, 31, 2, 5)) 
>>> b = np.random.random((9, 31, 2)) 
>>> a/b 
Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
ValueError: operands could not be broadcast together with shapes (9,31,2,5) (9,31,2) 

Answer 1

Как насчет

a = np.random.random((9, 31, 2, 5)) 
b = np.random.random((9, 31, 2, 1)) 
print((a/b).shape) 

урожайности

(9, 31, 2, 5)