taichi 三维模拟的输出数据转换.tfrecord格式

本人在研究GNN做流体运动模拟的时候参考 Sanchez-Gonzalez 的那篇GNS论文进行数据生成复现的时候,参考作者使用taichi library对流体运动用MPM方法进行仿真,输出数据怎么样才能转化成.tfrecord格式,用于后续对GNN的训练?看到文档里面有转化数据到numpy的方法,需要转化成numpy再转成TensorFlow的.tfrecord文件嘛?还是有什么简便的方法可以实现

.tfrecord 文件格式应该暂时没有支持,需要你自己用转换到 Numpy 等方法去搭建。关于转到 Numpy 的方法可以参考这里:Interacting with External Arrays | Taichi Docs