提交 22e414ff authored 作者: jordan-melendez's avatar jordan-melendez

Switched lambda to partial

上级 8aaf55bf
...@@ -4,6 +4,7 @@ import logging ...@@ -4,6 +4,7 @@ import logging
import warnings import warnings
import numpy as np import numpy as np
from six.moves import xrange from six.moves import xrange
from functools import partial
import theano import theano
from theano.tensor import as_tensor_variable from theano.tensor import as_tensor_variable
...@@ -375,10 +376,10 @@ class EighGrad(Op): ...@@ -375,10 +376,10 @@ class EighGrad(Op):
self.UPLO = UPLO self.UPLO = UPLO
if UPLO == 'L': if UPLO == 'L':
self.tri0 = np.tril self.tri0 = np.tril
self.tri1 = lambda a: np.triu(a, 1) self.tri1 = partial(np.triu, k=1)
else: else:
self.tri0 = np.triu self.tri0 = np.triu
self.tri1 = lambda a: np.tril(a, -1) self.tri1 = partial(np.tril, k=-1)
def make_node(self, x, w, v, gw, gv): def make_node(self, x, w, v, gw, gv):
x, w, v, gw, gv = map(as_tensor_variable, (x, w, v, gw, gv)) x, w, v, gw, gv = map(as_tensor_variable, (x, w, v, gw, gv))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论