Numpy: Wie Tensorprodukt richtig vektorisieren?
Hi, ich habe ein ndarray points von N × M Punkten in 3D (N, M, 3) und weiter ein ndarray rotations mit N Rotationsmatrizen (N, 3, 3).
Ich möchte nun für jedes N die M Punkte in 3D mit der entsprechenden Matrix rotieren, also:
for n in range(N):
points[n, :, :] = points[n] @ rotations[n].T
# (M, 3) = (M, 3) × (3, 3)
Nur eben ohne den Loop. Gibt es irgendwie einen Weg, das sauber in eine Tensoroperation zu stecken?