用taichi渲染sdf

我现在想用ray-marching可视化我用pytorch训练出来的一个sdf模型,想用taichi实现,我参考的是exsamples里的sdf_renderer.py,但我发现taichi的函数好像没法调用外部函数,我刚刚接触taichi,有人有什么好的建议吗

Taichi 有提供一部分基础的数学函数:Math Module | Taichi Docs
如果是更为复杂的外部函数,确实无法直接调用。有几个办法:1. 自己在 Taichi 中用 ti.kernel 来实现复杂函数的功能 2. 将 Taichi 中的数据转换成 Numpy Ndarray 等其他更为通用的数据格式,再交给外部函数,处理完了再搞回 ti.field ,暂时想到的就这些,其他如果还有办法请大家补充