提交 86f886f0 authored 作者: Olivier Breuleux's avatar Olivier Breuleux

nnet_ops -> nnet

上级 5ea17c2a
...@@ -2,7 +2,7 @@ import sys ...@@ -2,7 +2,7 @@ import sys
sys.path.insert(0, '..') sys.path.insert(0, '..')
import theano import theano
from theano import tensor as T from theano import tensor as T
from theano.tensor import nnet_ops from theano.tensor import nnet
from theano.compile import module from theano.compile import module
from theano import printing, pprint from theano import printing, pprint
from theano import compile from theano import compile
......
...@@ -485,7 +485,7 @@ class Canonizer(gof.LocalOptimizer): ...@@ -485,7 +485,7 @@ class Canonizer(gof.LocalOptimizer):
dcc += 1 dcc += 1
denum.remove(v) denum.remove(v)
denumct.append(ct) denumct.append(ct)
if self.use_reciprocal: if self.use_reciprocal or num:
ct = self.calculate(numct, denumct, aslist = True) ct = self.calculate(numct, denumct, aslist = True)
else: else:
ct = [self.calculate(numct, denumct, aslist = False)] ct = [self.calculate(numct, denumct, aslist = False)]
...@@ -540,7 +540,7 @@ def mul_calculate(num, denum, aslist = False): ...@@ -540,7 +540,7 @@ def mul_calculate(num, denum, aslist = False):
return [v] return [v]
return v return v
local_mul_canonizer = Canonizer(T.mul, T.div, T.inv, mul_calculate) local_mul_canonizer = Canonizer(T.mul, T.div, T.inv, mul_calculate, False)
@gof.local_optimizer([T.neg]) @gof.local_optimizer([T.neg])
def local_neg_to_mul(node): def local_neg_to_mul(node):
......
...@@ -6,7 +6,7 @@ from theano import gof ...@@ -6,7 +6,7 @@ from theano import gof
import test_basic as TT import test_basic as TT
import numpy import numpy
from theano.tensor.nnet_ops import * from theano.tensor.nnet import *
class T_sigmoid(unittest.TestCase): class T_sigmoid(unittest.TestCase):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论