Issue with Tensor Shape Mismatch in 3DAneurysm Script for Plotting Results in the HFM plot_results.m MATLAB File

[jjcrouch]

For the 3DAneurysm script, when attempting to plot the prediction results using the plot_results.m matlab file located in the data and figures folder linked on the HFM github here (https://o365coloradoedu-my.sharepoint.com/:f:/g/personal/mara4513_colorado_edu/EnMZcTjA6BVBjovGVJT3hk4BIemGxYF9lO6ry0SyizIjoQ?e=fuHKi2). I am receiving an error regarding the shape of the C, U, V, W, P_pred tensors saved to the dictionary in the predications.pkl file.

For example, using the pinnstorch Aneurysm3D.py script to save predictions, the tensor C_pred is of shape (98786, 1).
But, using the original tensorflow HFM Aneurysm3D.py script to save predictions, the tensor C_pred is of shape (98786, 301).
Unfortunately the plot_results.m file only works with a tensor of shape (98786, 301), this is because it matches the shape of the original 3DAneurysm.mat data file.

I am not a particularly high-level coder, so I am struggling to find where in the script I can adjust the pinnstorch 3DAneurysm script such that the model makes and saves a prediction on the entirety of the original .mat data file with shape (98786, 301). Any help would be greatly appreciated!