Additional modifications to allow for higher dimension response functions. Largely untested.

matlab_development/uq/toolbox/uq_pce_mean.m

-function [data, y] = uq_pce_mean(prob, data, u)
+function [data, y] = uq_pce_mean(prob, data, alphas)
 %UQ_PCE_MEAN Summary of this function goes here
 %  Given an array of PCE coefficients return the mean value of the
 %  expansion (first term in the expansion)
 
-y = u(1,:);
+alphas = reshape(alphas, data.Nt, []);
+y = alphas(1,:);
 
 end
\ No newline at end of file

matlab_development/uq/toolbox/uq_pce_mean_dU.m

-function [data, J] = uq_pce_mean_dU(prob, data, u)
+function [data, J] = uq_pce_mean_dU(prob, data, alphas)
 %UQ_PCE_MEAN Summary of this function goes here
 %  Given an array of PCE coefficients return the Jacobian of the mean value
 %  with respect to the expansion coefficients
-
-J = zeros(size(u'));
-J(1) = 1;
+alphas = reshape(alphas, data.Nt, []);
+J = zeros(size(alphas'));
+J(:,1) = 1;
 
 end

matlab_development/uq/toolbox/uq_pce_variance_dU.m

@@ -2,9 +2,9 @@ function [data, J] = uq_pce_variance_dU(prob, data, alphas)
 %UQ_PCE_VARIANCE_DU Jacobian of the variance of a PCE w/r/t alphas
 %  Calculate the Jacobian of the variance of a Polynomial Chaos Expansion
 %  (PCE) with respect to the PCE coefficients.
-
+alphas = reshape(alphas, data.Nt, []);
 J = 2*alphas';
-J(1) = 0;
+J(:,1) = 0;
 
 end