Векторизация в numpy.linalg.tensorsolve ⇐ Python

[Anonymous]

Я пытаюсь решить уравнение:
Ax = B
для x, где A и B — матрицы формы (3, 3, 3, 3) и (3, 3). Это можно сделать с помощью numpy.linalg.tensorsolve():

Код: Выделить всё

numpy.linalg.tensorsolve(A, B, axes=(0, 2))

Сейчас я пытаюсь решить это уравнение в n различных случаях, что означает мои матрицы A и B теперь имеют форму (10, 3, 3, 3, 3) и (10, 3, 3) соответственно.
Как мне действовать? об использовании tensorsolve в этом случае для векторизации моего кода? Я не знаю, какие оси указать и возможно ли это вообще.
Спасибо.

Подробнее здесь: https://stackoverflow.com/questions/665 ... ensorsolve