if * does mat-mat multiplication, the change is correct
The [] operator cannot be used at runtime with ints to get a subvector. At the specific position a matrix-matrix multiplication is needed. @claus-justus.heine Does the * operator for tensors contract over the last stufe?