提交 e13741a1 authored 作者: amrithasuresh's avatar amrithasuresh

1. Updated numpy as np

2. Fixed indentation
上级 8a58469a
from __future__ import absolute_import, print_function, division from __future__ import absolute_import, print_function, division
import numpy import numpy as np
import unittest import unittest
from theano.tests import unittest_tools as utt from theano.tests import unittest_tools as utt
import theano import theano
...@@ -43,13 +43,13 @@ class Test_inc_subtensor(unittest.TestCase): ...@@ -43,13 +43,13 @@ class Test_inc_subtensor(unittest.TestCase):
f = theano.function([a, increment, sl2_end], resut) f = theano.function([a, increment, sl2_end], resut)
val_a = numpy.ones((5, 5)) val_a = np.ones((5, 5))
val_inc = 2.3 val_inc = 2.3
val_sl2_end = 2 val_sl2_end = 2
result = f(val_a, val_inc, val_sl2_end) result = f(val_a, val_inc, val_sl2_end)
expected_result = numpy.copy(val_a) expected_result = np.copy(val_a)
if do_set: if do_set:
expected_result[:, :val_sl2_end] = val_inc expected_result[:, :val_sl2_end] = val_inc
else: else:
...@@ -71,7 +71,7 @@ class Test_inc_subtensor(unittest.TestCase): ...@@ -71,7 +71,7 @@ class Test_inc_subtensor(unittest.TestCase):
# These symbolic graphs legitimate, as long as increment has exactly # These symbolic graphs legitimate, as long as increment has exactly
# one element. So it should fail at runtime, not at compile time. # one element. So it should fail at runtime, not at compile time.
rng = numpy.random.RandomState(utt.fetch_seed()) rng = np.random.RandomState(utt.fetch_seed())
def rng_randX(*shape): def rng_randX(*shape):
return rng.rand(*shape).astype(theano.config.floatX) return rng.rand(*shape).astype(theano.config.floatX)
...@@ -101,7 +101,7 @@ class Test_inc_subtensor(unittest.TestCase): ...@@ -101,7 +101,7 @@ class Test_inc_subtensor(unittest.TestCase):
sl2 = slice(sl2_end) sl2 = slice(sl2_end)
sl3 = 2 sl3 = 2
val_a = numpy.ones((5, 3, 4)) val_a = np.ones((5, 3, 4))
val_inc = 2.3 val_inc = 2.3
val_sl2_end = 2 val_sl2_end = 2
...@@ -112,7 +112,7 @@ class Test_inc_subtensor(unittest.TestCase): ...@@ -112,7 +112,7 @@ class Test_inc_subtensor(unittest.TestCase):
f = theano.function([a, increment, sl2_end], resut) f = theano.function([a, increment, sl2_end], resut)
expected_result = numpy.copy(val_a) expected_result = np.copy(val_a)
result = f(val_a, val_inc, val_sl2_end) result = f(val_a, val_inc, val_sl2_end)
if method is tt.set_subtensor: if method is tt.set_subtensor:
...@@ -127,7 +127,7 @@ class Test_inc_subtensor(unittest.TestCase): ...@@ -127,7 +127,7 @@ class Test_inc_subtensor(unittest.TestCase):
f = theano.function([a, increment, sl2_end], resut) f = theano.function([a, increment, sl2_end], resut)
expected_result = numpy.copy(val_a) expected_result = np.copy(val_a)
result = f(val_a, val_inc, val_sl2_end) result = f(val_a, val_inc, val_sl2_end)
if method is tt.set_subtensor: if method is tt.set_subtensor:
...@@ -152,23 +152,23 @@ class Test_inc_subtensor(unittest.TestCase): ...@@ -152,23 +152,23 @@ class Test_inc_subtensor(unittest.TestCase):
# vector # vector
utt.verify_grad( utt.verify_grad(
f_slice(slice(2, 4, None)), f_slice(slice(2, 4, None)),
(numpy.asarray([0, 1, 2, 3, 4, 5.]), (np.asarray([0, 1, 2, 3, 4, 5.]),
numpy.asarray([9, 9.]), )) np.asarray([9, 9.]), ))
# matrix # matrix
utt.verify_grad( utt.verify_grad(
f_slice(slice(1, 2, None), slice(None, None, None)), f_slice(slice(1, 2, None), slice(None, None, None)),
(numpy.asarray([[0, 1], [2, 3], [4, 5.]]), (np.asarray([[0, 1], [2, 3], [4, 5.]]),
numpy.asarray([[9, 9.]]), )) np.asarray([[9, 9.]]), ))
# single element # single element
utt.verify_grad( utt.verify_grad(
f_slice(2, 1), f_slice(2, 1),
(numpy.asarray([[0, 1], [2, 3], [4, 5.]]), (np.asarray([[0, 1], [2, 3], [4, 5.]]),
numpy.asarray(9.),)) np.asarray(9.),))
# broadcast # broadcast
utt.verify_grad( utt.verify_grad(
f_slice(2), f_slice(2),
(numpy.asarray([[0, 1], [2, 3], [4, 5.]]), (np.asarray([[0, 1], [2, 3], [4, 5.]]),
numpy.asarray(9.),)) np.asarray(9.),))
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论