Hi,
I have 2D tensor of matrices, i.e I have matrices_tensor in size NxNx3x3 such that for index (i,j), matrices_tensor[i,j,…] yields matrix in size of 3x3.
I would like to do the following:
For target Nx3x1, I would like to get output in size Nx3 such that

output = torch.zeros((N,3))
for i in range(N):
for j in range(N):
output[i,:] += matrices_tensor[i,j] @ target[j,:]

where @ is matrix multiplication operator.
I want to do in tensor operation and not in loop in order to leverage the performance in gpu.
The matrix should be sparse, but for now, I’m looking for solution of dense matrix.
p.s:
It’s not necessary to know the application but if someone wants to know, the purpose is optimizing non rigid transformation for mesh deformation and regularize with smoothness of the rotated vertices.