提交 3753bad6 authored 作者: Frederic's avatar Frederic

pep8

上级 5bdeb8a3
...@@ -3660,8 +3660,8 @@ class T_Scan(unittest.TestCase): ...@@ -3660,8 +3660,8 @@ class T_Scan(unittest.TestCase):
#Bug reported by Bitton Tenessi #Bug reported by Bitton Tenessi
W_flat = tensor.fvector(name='W') W_flat = tensor.fvector(name='W')
W_flat.tag.test_value=numpy.ones((8,), dtype=numpy.float32) W_flat.tag.test_value = numpy.ones((8,), dtype=numpy.float32)
W = W_flat.reshape((2,2,2)) W = W_flat.reshape((2, 2, 2))
def loss_outer(i_outer, sum_outer, W): def loss_outer(i_outer, sum_outer, W):
...@@ -3669,20 +3669,22 @@ class T_Scan(unittest.TestCase): ...@@ -3669,20 +3669,22 @@ class T_Scan(unittest.TestCase):
return sum_inner + (W**2).sum().sum().sum() return sum_inner + (W**2).sum().sum().sum()
result_inner, _ = theano.scan(fn=loss_inner, result_inner, _ = theano.scan(
outputs_info = tensor.as_tensor_variable( fn=loss_inner,
numpy.asarray(0, dtype=numpy.float32)), outputs_info=tensor.as_tensor_variable(
sequences=tensor.arange(1, dtype='int32'), numpy.asarray(0, dtype=numpy.float32)),
non_sequences=[W], sequences=tensor.arange(1, dtype='int32'),
) non_sequences=[W],
)
return sum_outer + result_inner[-1] return sum_outer + result_inner[-1]
result_outer, _ = theano.scan(fn=loss_outer, result_outer, _ = theano.scan(
outputs_info = tensor.as_tensor_variable( fn=loss_outer,
numpy.asarray(0, dtype=numpy.float32)), outputs_info=tensor.as_tensor_variable(
sequences=tensor.arange(1, dtype='int32'), numpy.asarray(0, dtype=numpy.float32)),
non_sequences=[W], sequences=tensor.arange(1, dtype='int32'),
) non_sequences=[W],
)
cost = result_outer[-1] cost = result_outer[-1]
H = theano.gradient.hessian(cost, W_flat) H = theano.gradient.hessian(cost, W_flat)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论