Info<< "mesh size: " << mesh.C().size() << nl;
int cellNum = mesh.C().size();
vectorField uCell = U.field();
auto a = torch::zeros({cellNum, 2});
for (int i = 0; i < cellNum; i++)
{
a[i][0] = uCell[i].x();
a[i][1] = uCell[i].y();
};
std::cout<< "a:" << a << std::endl;
上面的代码可以把OpenFOAM的U场存储到a中