np.dot() с Python вещания

Question

У меня есть два Numpy массивов, один фасонных (3000,), а другой представляет собой массив из двадцати 3000 по 3000 матриц, то есть форма (20, 3000, 3000)

first.shape = (3000,) 
second.shape = (20, 3000, 3000) 

Я делаю numpy dot товар.

import numpy as np 
dotprod1 = np.dot(second, first) 

это работает, а выходной сигнал dotprod1 представляет собой массив в форме (20, 3000).

Но что, если я хочу снова взять точечный продукт?

dotprod2 = np.dot(first, dotprod1) 

Это дает нам ошибку.

ValueError: shapes (3000,) and (20,3000) not aligned: 3000 (dim 0) != 20 (dim 0) 

Я хотел бы иметь выход из 20 значений. Как использовать радиовещание для этого?

Answer 1

dotprod2 = np.dot(first, dotprod1) терпит неудачу, потому что first имеет форму (3000,) и dotprod1 имеет форму (20, 3000), поменять их местами и ошибка будет идти (если это ваше намерение):

dotprod2 = np.dot(dotprod1, first) 

кроме того, вы можете также использовать np.ndarray.dot сделать семантика ясно:

dotprod2 = dotprod1.dot(first)