Theano 简明教程
Theano - Computational Graph
从上述两个示例中,您可能已经注意到,在 Theano 中,我们创建一个表达式,该表达式最终使用 Theano function 进行评估。Theano 使用高级优化技术来优化表达式的执行。为了可视化计算图,Theano 在其库中提供了 printing 软件包。
Symbolic Graph for Scalar Addition
若要查看标量加法程序的计算图,请使用打印库如下:
theano.printing.pydotprint(f, outfile="scalar_addition.png", var_with_name_simple=True)
执行此语句后,将在您的机器上创建一个名为 scalar_addition.png 的文件。保存的计算图在此处显示,供您快速参考:
以下给出生成上述图像的完整程序清单:
from theano import *
a = tensor.dscalar()
b = tensor.dscalar()
c = a + b
f = theano.function([a,b], c)
theano.printing.pydotprint(f, outfile="scalar_addition.png", var_with_name_simple=True)
Symbolic Graph for Matrix Multiplier
现在,尝试为我们的矩阵乘法器创建计算图。生成此图的完整清单如下:
from theano import *
a = tensor.dmatrix()
b = tensor.dmatrix()
c = tensor.dot(a,b)
f = theano.function([a,b], c)
theano.printing.pydotprint(f, outfile="matrix_dot_product.png", var_with_name_simple=True)
生成的图形如下所示: