Трансляция точечного произведения списка 2D точек с матрицей 2x2

Question

У меня есть список точек xy с формой (2100).Я хочу взять скалярное произведение с матрицей 2x2 следующим образом:

g = xy.T@W@xy

, что должно привести к вектору 100 значений.Как я могу сделать это с Python?

Я знаю, что это должно привести к 100 значениям, потому что вышеупомянутый экспресс работает хорошо, если я кормлю в одной точке 2D.Как я могу векторизовать вышесказанное?

Answer 1

Мы можем np.einsum -

np.einsum('ij,ik,kj->j',xy,W,xy, optimize=True)