Умножение 2D-массива на 3D-массив векторов в NumpyPython

Программы на Python
Anonymous
Умножение 2D-массива на 3D-массив векторов в Numpy

Сообщение Anonymous »

У меня есть матрица dxd, которую я хотел бы умножить на матрицу dx1xn. Идея состоит в том, чтобы вернуть еще одну матрицу dx1xn, поэтому мне, по сути, хотелось бы умножить каждую матрицу dxd на каждую матрицу dx1 n раз вдоль этой третьей оси. Я понятия не имею, как это сделать. Я пробовал np.apply_along_axis и np.apply_over_axes, но первый выбирает только одну ось, а не матрицы dx1, а второй тоже не делает то, что я хочу.
k = 2
s = 36
d = k + s - 1
n = 1119

T = np.zeros(shape=(d, d)
alpha_filt = np.zeros(shape=(d, 1, n))

# Doesn't work obviously
np.matmul(T, alpha_filt)


Подробнее здесь: https://stackoverflow.com/questions/790 ... s-in-numpy

Вернуться в «Python»