Быстрый внешний тензорный продукт в NumPy
У меня есть два массива:
x of shape ((d1,...,d_m))
y of shape ((e_1,...e_n))
Я хотел бы сформировать внешнее тензорное произведение, то есть массив NumPy
z of shape ((d1,...,d_m,e_1,...,e_n))
такой, что
z[i_1,...,i_n,i_{n+1}...,i_{m+n}] == x[i_1,...i_m]*y[i_{m+1},...,i_{m+n}]
Я должен выполнить вышеупомянутое внешнее умножение несколько раз, поэтому я хотел бы максимально ускорить это.