您当前的位置:首页 >> 设计观点

胡渊鸣:import一个“太极”坎,让Python代码提速100倍

2023-04-19 12:16:05

uv[phase, i, j] lapl = uv[phase, i + 1, j] + uv[phase, i, j + 1] + uv[phase, i - 1, j] + uv[phase, i, j - 1] - 4.0 * cen du = Du * lapl[0] - cen[0] * cen[1] * cen[1] + feed * (1 - cen[0]) dv = Dv * lapl[1] + cen[0] * cen[1] * cen[1] - (feed + kill) * cen[1] val = cen + 0.5 * tm.vec2(du, dv) uv[1 - phase, i, j] = val

在这里,我们运用于整数相位(0或1)来控制我们从哪个交叉读取统计数据。

再一一步就是根据V的酸度对结果展开染色,就可以取得这样一个效果惊人的纹饰:

有趣的是,胡渊鸣介绍,即使V的初始酸度是随机设的,但每次都可以取得近似于的结果。

而且和不能降到30fps左右的Numba借助比好像,Taichi借助由于可以选项GPU作为后尾端,平易近人超过了 300fps。

pip install即可安装

看剩年前面三个例三子,你这下相信了吧?

其实,Taichi就是一个嵌入在Python里的DSL(自适应应用程序员),它通过自己的解释器器将被 @ti.kernel 花纹的formula_解释器到各种硬件上,都有CPU和GPU,然后展开小型化计数。

有了它,你须要再羡慕C++/CUDA的性能指标。

正如其名,Taichi就出自实在极绘图胡渊鸣的团队,现在你只需要用pip install就能安装这个戈,并与其他Python戈展开交互,都有NumPy、Matplotlib和PyTorch等等。

当然,Taichi用好像和这些戈以及其他减速方法有什么差别,胡渊鸣也说明了简要的种设计对比,感兴趣的好友可以戳后面的链接简要查看:

— 剩 —

量三子位 QbitAI · 头条号经理人

关注我们,第一一段时间获知年前沿科技自适应

广州白癜风医院专家预约挂号
用什么药能缓解腰背痛
郑州白癜风医院哪家治疗最好
咸阳白癜风医院
消化不良的调理方法
相关阅读
友情链接