Simulink 3D Matrix Multiply

Hi, I have a set of state-space equations that I'm trying to simultaneously propagate. The A matrix will be an nxnxm matrix and the state matrix will be a nxm matrix.

In Matlab code, this function would look like:

for(i = 1:m)

   Uprime(:,i) = A(:,:,i)*U(:,i) + b;
end

What is the best way to implement this in Simulink? I was thinking of writing an s-function, to loop through the third dimension, but I was hoping there was a built in Simulink function that could do this instead?