在代码中，vtk['pred'] = u_pred[:,0] 的问题在于 u_pred[:,0] 的形状仍然是 (1, numpoint)，而不是 (numpoint,)。可以通过以下方式修复 u_pred_squeezed = np.squeeze(u_pred) vtk['pred'] = u_pred_squeezed
在代码中，vtk['pred'] = u_pred[:,0] 的问题在于 u_pred[:,0] 的形状仍然是 (1, numpoint)，而不是 (numpoint,)。可以通过以下方式修复
u_pred_squeezed = np.squeeze(u_pred)
vtk['pred'] = u_pred_squeezed