Theano是一个Python库,可以让你来定义、优化和评估数学计算表达式,尤其是多维数组。使用Theano在处理大数据时可能达到和纯用C的运算速度。它能通过利用多个GPU的优势超越用C在CPU上的运行。
。。。
Theano优化包括但不仅限于:
- 使用GPU计算
- 常量分析
- 合并相似子图,避免冗余计算
- 算术简化
- 插入有效的BLAS操作
- 使用内存混叠避免计算
- 采用inplace操作
- 循环融合
一睹为快
下面是一个如何使用Theano的例子,尽管不能包含其全部特点,但它详细阐述了如何使用Theano:
import theano
from theano import tensor
# declare two symbolic floating-point scalars
a = tensor.dscalar()
b = tensor.dscalar()
# create a simple expression
c = a + b
# convert the expression into a callable object that takes (a,b)
# values as input and computes a value for c
f = theano.function([a,b], c)
# bind 1.5 to 'a', 2.5 to 'b', and evaluate 'c'
assert 4.0 == f(1.5, 2.5)