关键词：
丰色 发自 凹非寺
量子位 | 公众号 QbitAI
众所周知，Python的简单和易读性是靠牺牲性能为代价的——
尤其是在计算密集的情况下，比如多重for循环。
不过现在，大佬胡渊鸣说了：
只需import 一个叫做“Taichi”的库，就可以把代码速度提升100倍！
不信？
来看三个例子。
计算素数的个数，速度x120
第一个例子非常非常简单，求所有小于给定正整数N的素数。
标准答案如下：
我们将上面的代码保存，运行。
当N为100万时，需要2.235s得到结果：
现在，我们开始施魔法。
不用更改任何函数体，import“taichi”库，然后再加两个装饰器：
Bingo！同样的结果只要0.363s，快了将近6倍。
如果N=1000万，则只要0.8s；要知道，不加它可是55s，一下子又快了70倍！
不止如此，我们还可以在ti.init()中加个参数变为ti.init(arch=ti.gpu) ，让taich在GPU上进行计算。
那么此时，计算所有小于1000万的素数就只耗时0.45s了，与原来的Python代码相比速度就提高了120倍！
厉不厉害？
什么？你觉得这个例子太简单了，说服力不够？我们再来看一个稍微复杂一点的。
动态规划，速度x500
动态规划不用多说，作为一种优化算法，通过动态存储中间计算结果来减少计算时间。
我们以经典教材《算法导论》中的经典动态规划案例“最长公共子序列问题（LCS）”为例。
比如对于序列a = [0, 1, 0, 2, 4, 3, 1, 2, 1]和序列b = [4, 0, 1, 4, 5, 3, 1, 2]，它们的LCS就是：
LCS(a, b) = [0, 1, 4, 3, 1, 2]。
用动态规划的思路计算LCS，就是先求解序列a的前i个元素和序列b的前j个元素的最长公共子序列的长度，然后逐步增加i或j的值，重复过程，得到结果。
我们用f[i, j]来指代这个子序列的长度，即LCS((prefix(a, i), prefix(b, j)。其中prefix(a, i) 表示序列a的前i个元素，即a[0], a[1], …, a[i - 1]，得到如下递归关系：
完整代码如下：
现在，我们用Taichi来加速：
结果如下：
胡渊鸣电脑上的程序最快做到了0.9秒内完成，而换成用NumPy来实现，则需要476秒，差异达到了超500倍！
最后，我们再来一个不一样的例子。
反应 - 扩散方程，效果惊人
自然界中，总有一些动物身上长着一些看起来无序但实则并非完全随机的花纹。
图灵机的发明者艾伦·图灵是第一个提出模型来描述这种现象的人。
在该模型中，两种化学物质（U和V）来模拟图案的生成。这两者之间的关系类似于猎物和捕食者，它们自行移动并有交互：
最初，U和V随机分布在一个域上；
在每个时间步，它们逐渐扩散到邻近空间；
当U和V相遇时，一部分U被V吞噬。因此，V的浓度增加；
为了避免U被V根除，我们在每个时间步添加一定百分比 (f) 的U并删除一定百分比 (k) 的V。
上面这个过程被概述为“反应-扩散方程”：
其中有四个关键参数：Du（U的扩散速度），Dv（V的扩散速度），f（feed的缩写，控制U的加入）和k（kill的缩写，控制V的去除）。
如果Taichi中实现这个方程，首先创建网格来表示域，用vec2表示每个网格中U, V的浓度值。
拉普拉斯算子数值的计算需要访问相邻网格。为了避免在同一循环中更新和读取数据，我们应该创建两个形状相同的网格W×H×2。
每次从一个网格访问数据时，我们将更新的数据写入另一个网格，然后切换下一个网格。那么数据结构设计就是这样：
一开始，我们将U在网格中的浓度设置为 1，并将V放置在50个随机选择的位置：
那么实际计算就可以用不到10行代码完成：
@ti.kernel
def compute(phase: int):
for i, j in ti.ndrange(W, H):
cen = 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 装饰的函数编译到各种硬件上，包括CPU和GPU，然后进行高性能计算。
有了它，你无需再羡慕C++/CUDA的性能。
正如其名，Taichi就出自太极图形胡渊鸣的团队，现在你只需要用pip install就能安装这个库，并与其他Python库进行交互，包括NumPy、Matplotlib和PyTorch等等。
当然，Taichi用起来和这些库以及其他加速方法有什么差别，胡渊鸣也给出了详细的优缺点对比，感兴趣的朋友可以戳下面的链接详细查看：
https://docs.taichi-lang.org/blog/accelerate-python-code-100x
— 完 —
import一个“太极”库，让python代码提速100倍！
...集的情况下，比如多重for循环。不过现在，大佬胡渊鸣说了：只需import一个叫做“Taichi”的库，就可以把代码速度提升100倍！不信？来看三个例子。计算素数的个数，速度 查看详情
2013集训胡渊鸣城市规划(代码片段)
题面Description刚刚解决完电力网络的问题,阿狸又被领导的任务给难住了. 刚才说过,阿狸的国家有n个城市,现在国家需要在某些城市对之间建立一些贸易路线,使得整个国家的任意两个城市都直接或间接的连通.为了省钱,每两个... 查看详情
新特效编程语言-taichi太极
.../finance.sina.com.cn/wm/2020-01-06/doc-iihnzahk2279095.shtml） 胡渊鸣知乎原文：https://zhuanlan.zhihu.com/p/97700605论文地址：https://arxiv.org/abs/1910.00935太极项目地址：https://github.com/yuanming-hu/taichitaichi_mpm项目地址:https://github.com/yuanming-hu/taich... 查看详情
转：新特效编程语言-taichi太极
.../finance.sina.com.cn/wm/2020-01-06/doc-iihnzahk2279095.shtml） 胡渊鸣知乎原文：https://zhuanlan.zhihu.com/p/97700605论文地址：https://arxiv.org/abs/1910.00935太极项目地址：https://github.com/yuanming-hu/taichitaichi_mpm项目地址:https://github.com/yuanming-hu/taich... 查看详情
图形计算好玩的taichi示例展示(代码片段)
...下taichi的安装方法及示例演示。 taichi也称太极，是胡渊鸣大佬的作品，是专门为高性能计算机图形学设计的编程语言，它深深地嵌入在python中，并且它的即时编译器将计算密集型任务转移到多核CPU和大规模并行G... 查看详情
import详解(代码片段)
...hon解释器能够搜索到这个模块，这种方式比较灵活。查看import搜索路径importsysforiinsys.path:print(i)设置搜索路径im 查看详情
python调用问题，求解
...问为什么第三方库可以直接调用Bar()这个类？书中前面只importpygal，将第三方库导入。运行起来正常出结果但是为什么可以这么调用？不应该是from模块import类吗？参考技术A这个只是导入的方式不同。可以把库看作一个工具箱（比... 查看详情
python中标准模块importlib详解(代码片段)
Python中标准模块importlib详解模块简介Python提供了importlib包作为标准库的一部分。目的就是提供Python中import语句的实现（以及__import__函数）。另外，importlib允许程序员创建他们自定义的对象，可用于引入过程（也称为importer）。什... 查看详情
Python3 - 无法在“import cairosvg”上加载库
】Python3-无法在“importcairosvg”上加载库【英文标题】：Python3-Failedtoloadlibraryon"importcairosvg"【发布时间】：2019-03-0521:21:54【问题描述】：我正在尝试为Python3安装CairoSVG。我运行pip3installcairosvg没有任何错误，但是当我尝试通... 查看详情
python库、包、模块概念辨析
...可以导入这些模块（.py文件）。模块方式：1、from模块名import函数名2、import模块名可以使用as为模块或函数起一个别名包在模块之上的概念，为了方便管理而将.py文件进行打包。包目录下第一个文件便是init.py(特点)，然后是一些... 查看详情
python标准库和第三方库有啥区别
...及使用方法不同。3、它们调用方式是一样的，都需要用import语句调用。简单的说，一个是默认自带不需要下载安装的库，一个是需要下载安装的库。它们的调用方式是一样的。参考技术A它们的主要区别是：1、Python的标准库是随... 查看详情
太极旋转-js实现
刚学了js的一些函数，所以做了一个太极的旋转。做完之后是上面这个样子的，是可以旋转的。 思路： 1.先做一个基准转盘，之后将元素都放在转盘上，跟随转盘动。 2.画两个半圆，主要属性是border-top-right-radius:... 查看详情
如何在 azure ML 中导入某些 python 库？就像 import humanfriendly 行一样会出错
】如何在azureML中导入某些python库？就像importhumanfriendly行一样会出错【英文标题】：HowcancertainpythonlibrariesbeimportedinazureML?Likethelineimporthumanfriendlygiveserror【发布时间】：2017-11-1911:31:40【问题描述】：在执行python模块。线importhumanfrie... 查看详情
import的使用(代码片段)
首先来看一个例子：importmathprint(math.pi)#输出3.141592653589793上面的程序使用了import语句，importmath的意思是从Python标准库中引入math.py模块，这是Python中定义的引入模块的方法可以用如下方法引入多个模块：importmath,time,sys除了用import... 查看详情
如何让 Sass Importer 将文件合并到一个样式表中？ [复制]
】如何让SassImporter将文件合并到一个样式表中？[复制]【英文标题】：HowdoImaketheSassImportercombinefilesintoonestylesheet?[duplicate]【发布时间】：2015-11-0909:08:56【问题描述】：我最近在名为importer的node-sass中发现了这个功能，它允许您在sa... 查看详情
sublime中使用不了python自定义包
...行的。Sublime安装也没问题；可以运行HelloWorld。那么怎样Import自定义包，怎样才有正确输出？参考技术A单击你上边sublime截图右下角的PlainText,选择python，Ctrl+B执行追问还是没有成功。见下图。追答你不会是装了sub什么也没设置吧... 查看详情
cmake 将外部库与 IMPORT_SONAME ro IMPORT_LOCATION 链接
】cmake将外部库与IMPORT_SONAMEroIMPORT_LOCATION链接【英文标题】：cmaketolinkexternallibrarywithIMPORT_SONAMEroIMPORT_LOCATION【发布时间】：2021-06-2814:12:31【问题描述】：我有一个链接到外部库的C++项目该库由供应商提供，其中仅包含一个.h标头... 查看详情
模块的使用import，fromimport的使用
...成功的python文件，列如module.py，模块名是module#可以使用importmodule，#四个通用类别：#1使用python编写的.py文件#2已被编译为共享库或DLL的C或C++扩展#3把一系列模块组织到一 查看详情