提交 eca91d87 authored 作者: Olivier Breuleux's avatar Olivier Breuleux

added gemm pattern

上级 84a66c9f
......@@ -4,6 +4,17 @@ from elemwise import Elemwise, DimShuffle
import scalar
gemm_pattern_1 = theano.gof.PatternSub((T.sub_inplace,
'd',
(T.mul,
dict(pattern = (T.DimShuffle((), ['x', 'x'], inplace = True), 'a'),
allow_multiple_clients = True),
(T.dot, 'b', 'c'))),
(T.gemm, 'd', 'a', 'b', 'c', T.constant(-1.0)),
allow_multiple_clients = False)
class InplaceOptimizer(gof.Optimizer):
"""
Usage: inplace_optimizer.optimize(env)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论