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

pep8

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