I am interested in calculating a function on a bunch of permutations of parameter values. I want to keep it generic to N dimensions, but let me write it out in 3 dimensions to start with. Generating the permutations is easy enough with meshgrid
, but I can't figure out how to reshape the resulting array back to the multidimensions? Here is a starting point:
%These are the 3 variations of parameters, with some values.
params1 = [100, 200, 300];%Picking these so it is easy to correlate to the function
params2 = [10, 20];
params3 = [1, 2];
%This generates parameter_values as the cartesian productpermutations.
[vec1, vec2, vec3] = meshgrid(params1, params2, params3);
parameter_values = [vec1(:) vec2(:) vec3(:)];
%Calculates functions on the set of parameters.
%Would have a fancier function, of course, this just makes it easy to see the results.
raw_vals = parameter_values(:,1) + parameter_values(:,2) + parameter_values(:,3);
%Rearrange into a multiarray to access by parameter indices.
f_vals = reshape(raw_vals, [length(params1), length(params2), length(params3)]) %WRONG?
%THE FOLLOWING FAIL BUT WOULD BE EXPECTED WITH THESE PARAMETERS AND THE FUNCTION.
assert(f_vals(2,1,1) == 211)
assert(f_vals(3,2,2) == 322)
You want
ndgrid
instead ofmeshgrid
in this case.meshgrid
's syntax is[X,Y] = meshgrid(xgv,ygv)
which causesY(:)
to vary fastest rather thanX(:)
. See Gridded Data Representation for more details. In other words, you are gettingBut you want to be getting:
If you switch to
ndgrid
, then you getf_vals(2,1,1) == 211
as intended.Generalizing to N-dimensions could be done like this: