Skip to content
项目
群组
代码片段
帮助
当前项目
正在载入...
登录 / 注册
切换导航面板
P
pytensor
项目
项目
详情
活动
周期分析
仓库
仓库
文件
提交
分支
标签
贡献者
图表
比较
统计图
议题
0
议题
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
CI / CD
CI / CD
流水线
作业
日程
统计图
Wiki
Wiki
代码片段
代码片段
成员
成员
折叠边栏
关闭边栏
活动
图像
聊天
创建新问题
作业
提交
问题看板
Open sidebar
testgroup
pytensor
Commits
578eb363
提交
578eb363
authored
3月 21, 2008
作者:
turian@grenat.iro.umontreal.ca
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Renamed tinit to astensor
上级
a68aa7e2
隐藏空白字符变更
内嵌
并排
正在显示
3 个修改的文件
包含
59 行增加
和
60 行删除
+59
-60
_test_elemwise.py
_test_elemwise.py
+4
-4
_test_tensor.py
_test_tensor.py
+52
-53
tensor.py
tensor.py
+3
-3
没有找到文件。
_test_elemwise.py
浏览文件 @
578eb363
...
@@ -2,7 +2,7 @@
...
@@ -2,7 +2,7 @@
import
unittest
import
unittest
import
numpy
import
numpy
from
tensor
import
tinit
,
Tensor
from
tensor
import
astensor
,
Tensor
import
gof
import
gof
from
gof
import
modes
,
Env
from
gof
import
modes
,
Env
...
@@ -29,9 +29,9 @@ def inputs():
...
@@ -29,9 +29,9 @@ def inputs():
l1
=
[[
1.0
,
2.0
],
[
3.0
,
4.0
]]
l1
=
[[
1.0
,
2.0
],
[
3.0
,
4.0
]]
l2
=
[[
3.0
,
4.0
],
[
1.0
,
2.0
]]
l2
=
[[
3.0
,
4.0
],
[
1.0
,
2.0
]]
l3
=
numpy
.
ones
((
2
,
3
))
l3
=
numpy
.
ones
((
2
,
3
))
x
=
modes
.
build
(
tinit
(
l1
,
name
=
'x'
))
x
=
modes
.
build
(
astensor
(
l1
,
name
=
'x'
))
y
=
modes
.
build
(
tinit
(
l2
,
name
=
'y'
))
y
=
modes
.
build
(
astensor
(
l2
,
name
=
'y'
))
z
=
modes
.
build
(
tinit
(
l3
,
name
=
'z'
))
z
=
modes
.
build
(
astensor
(
l3
,
name
=
'z'
))
return
x
,
y
,
z
return
x
,
y
,
z
def
env
(
inputs
,
outputs
,
validate
=
True
,
features
=
[]):
def
env
(
inputs
,
outputs
,
validate
=
True
,
features
=
[]):
...
...
_test_tensor.py
浏览文件 @
578eb363
...
@@ -18,7 +18,7 @@ def verify_grad(testcase, op_cls, pt, n_tests=1, rng=numpy.random, eps=0.0000001
...
@@ -18,7 +18,7 @@ def verify_grad(testcase, op_cls, pt, n_tests=1, rng=numpy.random, eps=0.0000001
pt
=
[
numpy
.
asarray
(
p
)
for
p
in
pt
]
pt
=
[
numpy
.
asarray
(
p
)
for
p
in
pt
]
for
test_num
in
xrange
(
n_tests
):
for
test_num
in
xrange
(
n_tests
):
tensor_pt
=
[
tinit
(
p
,
name
=
'input
%
i'
%
i
)
for
i
,
p
in
enumerate
(
pt
)]
tensor_pt
=
[
astensor
(
p
,
name
=
'input
%
i'
%
i
)
for
i
,
p
in
enumerate
(
pt
)]
o
=
op_cls
(
*
tensor_pt
)
o
=
op_cls
(
*
tensor_pt
)
if
len
(
o
.
outputs
)
>
1
:
if
len
(
o
.
outputs
)
>
1
:
raise
NotImplementedError
(
'cant (yet) autotest gradient of op with multiple outputs'
)
raise
NotImplementedError
(
'cant (yet) autotest gradient of op with multiple outputs'
)
...
@@ -28,7 +28,7 @@ def verify_grad(testcase, op_cls, pt, n_tests=1, rng=numpy.random, eps=0.0000001
...
@@ -28,7 +28,7 @@ def verify_grad(testcase, op_cls, pt, n_tests=1, rng=numpy.random, eps=0.0000001
o_fn
=
Function
(
tensor_pt
,
o
.
outputs
)
o_fn
=
Function
(
tensor_pt
,
o
.
outputs
)
o_fn_out
=
o_fn
(
*
pt
)
o_fn_out
=
o_fn
(
*
pt
)
random_projection
=
rng
.
rand
(
*
o_fn_out
.
shape
)
random_projection
=
rng
.
rand
(
*
o_fn_out
.
shape
)
t_r
=
tinit
(
random_projection
)
t_r
=
astensor
(
random_projection
)
#random projection of o onto t_r
#random projection of o onto t_r
cost
=
sum
(
t_r
*
o
.
outputs
[
0
])
cost
=
sum
(
t_r
*
o
.
outputs
[
0
])
...
@@ -36,7 +36,7 @@ def verify_grad(testcase, op_cls, pt, n_tests=1, rng=numpy.random, eps=0.0000001
...
@@ -36,7 +36,7 @@ def verify_grad(testcase, op_cls, pt, n_tests=1, rng=numpy.random, eps=0.0000001
num_grad
=
gradient
.
numeric_grad
(
cost_fn
,
pt
)
num_grad
=
gradient
.
numeric_grad
(
cost_fn
,
pt
)
symbolic_grad
=
gradient
.
grad
(
cost
,
tensor_pt
,
tinit
(
1.0
,
name
=
'g_cost'
))
symbolic_grad
=
gradient
.
grad
(
cost
,
tensor_pt
,
astensor
(
1.0
,
name
=
'g_cost'
))
if
0
:
if
0
:
print
'-------'
print
'-------'
print
'----------'
print
'----------'
...
@@ -74,47 +74,47 @@ class T_argmax(unittest.TestCase):
...
@@ -74,47 +74,47 @@ class T_argmax(unittest.TestCase):
Argmax
.
debug
=
0
Argmax
.
debug
=
0
def
test0
(
self
):
def
test0
(
self
):
n
=
tinit
(
5.0
)
n
=
astensor
(
5.0
)
v
,
i
=
eval_outputs
(
argmax
(
n
))
v
,
i
=
eval_outputs
(
argmax
(
n
))
self
.
failUnless
(
v
==
5.0
)
self
.
failUnless
(
v
==
5.0
)
self
.
failUnless
(
i
==
0
)
self
.
failUnless
(
i
==
0
)
def
test1
(
self
):
def
test1
(
self
):
n
=
tinit
([
1
,
2
,
3
,
2
,
-
6
])
n
=
astensor
([
1
,
2
,
3
,
2
,
-
6
])
v
,
i
=
eval_outputs
(
argmax
(
n
))
v
,
i
=
eval_outputs
(
argmax
(
n
))
self
.
failUnless
(
v
==
3
)
self
.
failUnless
(
v
==
3
)
self
.
failUnless
(
i
==
2
)
self
.
failUnless
(
i
==
2
)
def
test2
(
self
):
def
test2
(
self
):
n
=
tinit
(
numpy
.
random
.
rand
(
2
,
3
))
n
=
astensor
(
numpy
.
random
.
rand
(
2
,
3
))
v
,
i
=
eval_outputs
(
argmax
(
n
))
v
,
i
=
eval_outputs
(
argmax
(
n
))
self
.
failUnless
(
numpy
.
all
(
i
==
[
0
,
1
]))
self
.
failUnless
(
numpy
.
all
(
i
==
[
0
,
1
]))
def
test2b
(
self
):
def
test2b
(
self
):
n
=
tinit
(
numpy
.
random
.
rand
(
2
,
3
))
n
=
astensor
(
numpy
.
random
.
rand
(
2
,
3
))
v
,
i
=
eval_outputs
(
argmax
(
n
,
axis
=
0
))
v
,
i
=
eval_outputs
(
argmax
(
n
,
axis
=
0
))
self
.
failUnless
(
numpy
.
all
(
i
==
[
0
,
1
,
1
]))
self
.
failUnless
(
numpy
.
all
(
i
==
[
0
,
1
,
1
]))
def
test2_invalid
(
self
):
def
test2_invalid
(
self
):
n
=
tinit
(
numpy
.
random
.
rand
(
2
,
3
))
n
=
astensor
(
numpy
.
random
.
rand
(
2
,
3
))
try
:
try
:
eval_outputs
(
argmax
(
n
,
axis
=
3
))
eval_outputs
(
argmax
(
n
,
axis
=
3
))
self
.
fail
()
self
.
fail
()
except
ValueError
,
e
:
except
ValueError
,
e
:
return
return
def
test2_invalid_neg
(
self
):
def
test2_invalid_neg
(
self
):
n
=
tinit
(
numpy
.
random
.
rand
(
2
,
3
))
n
=
astensor
(
numpy
.
random
.
rand
(
2
,
3
))
try
:
try
:
eval_outputs
(
argmax
(
n
,
axis
=-
3
))
eval_outputs
(
argmax
(
n
,
axis
=-
3
))
self
.
fail
()
self
.
fail
()
except
ValueError
,
e
:
except
ValueError
,
e
:
return
return
def
test2_valid_neg
(
self
):
def
test2_valid_neg
(
self
):
n
=
tinit
(
numpy
.
random
.
rand
(
2
,
3
))
n
=
astensor
(
numpy
.
random
.
rand
(
2
,
3
))
v
,
i
=
eval_outputs
(
argmax
(
n
,
axis
=-
1
))
v
,
i
=
eval_outputs
(
argmax
(
n
,
axis
=-
1
))
self
.
failUnless
(
v
.
shape
==
(
2
,))
self
.
failUnless
(
v
.
shape
==
(
2
,))
v
,
i
=
eval_outputs
(
argmax
(
n
,
axis
=-
2
))
v
,
i
=
eval_outputs
(
argmax
(
n
,
axis
=-
2
))
self
.
failUnless
(
v
.
shape
==
(
3
,))
self
.
failUnless
(
v
.
shape
==
(
3
,))
def
test3
(
self
):
def
test3
(
self
):
n
=
tinit
(
numpy
.
random
.
rand
(
2
,
3
,
4
))
n
=
astensor
(
numpy
.
random
.
rand
(
2
,
3
,
4
))
v
,
i
=
eval_outputs
(
argmax
(
n
,
axis
=
0
))
v
,
i
=
eval_outputs
(
argmax
(
n
,
axis
=
0
))
self
.
failUnless
(
v
.
shape
==
(
3
,
4
))
self
.
failUnless
(
v
.
shape
==
(
3
,
4
))
self
.
failUnless
(
i
.
shape
==
(
3
,
4
))
self
.
failUnless
(
i
.
shape
==
(
3
,
4
))
...
@@ -128,7 +128,7 @@ class T_argmax(unittest.TestCase):
...
@@ -128,7 +128,7 @@ class T_argmax(unittest.TestCase):
class
T_transpose
(
unittest
.
TestCase
):
class
T_transpose
(
unittest
.
TestCase
):
def
test0
(
self
):
def
test0
(
self
):
n
=
tinit
(
numpy
.
ones
(()))
n
=
astensor
(
numpy
.
ones
(()))
t
=
transpose
(
n
)
t
=
transpose
(
n
)
self
.
failUnless
(
t
.
owner
.
__class__
is
Transpose
)
self
.
failUnless
(
t
.
owner
.
__class__
is
Transpose
)
f
=
Function
([
n
],
[
t
])
f
=
Function
([
n
],
[
t
])
...
@@ -140,7 +140,7 @@ class T_transpose(unittest.TestCase):
...
@@ -140,7 +140,7 @@ class T_transpose(unittest.TestCase):
self
.
failUnless
(
n
.
data
==
56.0
)
self
.
failUnless
(
n
.
data
==
56.0
)
def
test1
(
self
):
def
test1
(
self
):
n
=
tinit
(
numpy
.
ones
(
5
))
n
=
astensor
(
numpy
.
ones
(
5
))
t
=
transpose
(
n
)
t
=
transpose
(
n
)
self
.
failUnless
(
t
.
owner
.
__class__
is
Transpose
)
self
.
failUnless
(
t
.
owner
.
__class__
is
Transpose
)
f
=
Function
([
n
],
[
t
])
f
=
Function
([
n
],
[
t
])
...
@@ -151,7 +151,7 @@ class T_transpose(unittest.TestCase):
...
@@ -151,7 +151,7 @@ class T_transpose(unittest.TestCase):
self
.
failUnless
(
n
.
data
[
0
]
==
56.0
)
self
.
failUnless
(
n
.
data
[
0
]
==
56.0
)
def
test2
(
self
):
def
test2
(
self
):
n
=
tinit
(
numpy
.
ones
((
5
,
3
)))
n
=
astensor
(
numpy
.
ones
((
5
,
3
)))
t
=
transpose
(
n
)
t
=
transpose
(
n
)
self
.
failUnless
(
t
.
owner
.
__class__
is
Transpose
)
self
.
failUnless
(
t
.
owner
.
__class__
is
Transpose
)
f
=
Function
([
n
],
[
t
])
f
=
Function
([
n
],
[
t
])
...
@@ -162,7 +162,7 @@ class T_transpose(unittest.TestCase):
...
@@ -162,7 +162,7 @@ class T_transpose(unittest.TestCase):
self
.
failUnless
(
n
.
data
[
0
,
0
]
==
56.0
)
self
.
failUnless
(
n
.
data
[
0
,
0
]
==
56.0
)
def
test3
(
self
):
def
test3
(
self
):
n
=
tinit
(
numpy
.
ones
((
5
,
3
,
2
)))
n
=
astensor
(
numpy
.
ones
((
5
,
3
,
2
)))
t
=
transpose
(
n
)
t
=
transpose
(
n
)
self
.
failUnless
(
t
.
owner
.
__class__
is
Transpose
)
self
.
failUnless
(
t
.
owner
.
__class__
is
Transpose
)
f
=
Function
([
n
],
[
t
])
f
=
Function
([
n
],
[
t
])
...
@@ -175,14 +175,14 @@ class T_transpose(unittest.TestCase):
...
@@ -175,14 +175,14 @@ class T_transpose(unittest.TestCase):
class
T_subtensor
(
unittest
.
TestCase
):
class
T_subtensor
(
unittest
.
TestCase
):
def
test0_err_invalid
(
self
):
def
test0_err_invalid
(
self
):
#it is impossible to retrieve a view of a 0-d tensor
#it is impossible to retrieve a view of a 0-d tensor
n
=
tinit
(
numpy
.
ones
(()))
n
=
astensor
(
numpy
.
ones
(()))
try
:
try
:
t
=
n
[
0
]
t
=
n
[
0
]
self
.
fail
()
self
.
fail
()
except
ValueError
,
e
:
except
ValueError
,
e
:
self
.
failUnless
(
e
[
0
]
is
Subtensor
.
e_invalid
)
self
.
failUnless
(
e
[
0
]
is
Subtensor
.
e_invalid
)
def
test1_err_bounds
(
self
):
def
test1_err_bounds
(
self
):
n
=
tinit
(
numpy
.
ones
(
3
))
n
=
astensor
(
numpy
.
ones
(
3
))
t
=
n
[
7
]
t
=
n
[
7
]
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
try
:
try
:
...
@@ -192,42 +192,42 @@ class T_subtensor(unittest.TestCase):
...
@@ -192,42 +192,42 @@ class T_subtensor(unittest.TestCase):
if
e
[
0
]
!=
'index out of bounds'
:
if
e
[
0
]
!=
'index out of bounds'
:
raise
raise
def
test1_ok_range_finite
(
self
):
def
test1_ok_range_finite
(
self
):
n
=
tinit
(
numpy
.
ones
(
3
)
*
5
)
n
=
astensor
(
numpy
.
ones
(
3
)
*
5
)
t
=
n
[
0
:
2
]
t
=
n
[
0
:
2
]
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
tval
=
eval_outputs
([
t
])
tval
=
eval_outputs
([
t
])
self
.
failUnless
(
tval
.
shape
==
(
2
,))
self
.
failUnless
(
tval
.
shape
==
(
2
,))
self
.
failUnless
(
tval
[
1
]
==
5.0
)
self
.
failUnless
(
tval
[
1
]
==
5.0
)
def
test2_ok_range_finite
(
self
):
def
test2_ok_range_finite
(
self
):
n
=
tinit
(
numpy
.
ones
((
3
,
4
))
*
5
)
n
=
astensor
(
numpy
.
ones
((
3
,
4
))
*
5
)
t
=
n
[
0
:
2
,
3
]
t
=
n
[
0
:
2
,
3
]
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
tval
=
eval_outputs
([
t
])
tval
=
eval_outputs
([
t
])
self
.
failUnless
(
tval
.
shape
==
(
2
,))
self
.
failUnless
(
tval
.
shape
==
(
2
,))
self
.
failUnless
(
tval
[
1
]
==
5.0
)
self
.
failUnless
(
tval
[
1
]
==
5.0
)
def
test1_err_invalid
(
self
):
def
test1_err_invalid
(
self
):
n
=
tinit
(
numpy
.
ones
(
1
))
n
=
astensor
(
numpy
.
ones
(
1
))
try
:
try
:
t
=
n
[
0
,
0
]
t
=
n
[
0
,
0
]
self
.
fail
()
self
.
fail
()
except
ValueError
,
e
:
except
ValueError
,
e
:
self
.
failUnless
(
e
[
0
]
is
Subtensor
.
e_invalid
)
self
.
failUnless
(
e
[
0
]
is
Subtensor
.
e_invalid
)
def
test1_ok_elem
(
self
):
def
test1_ok_elem
(
self
):
n
=
tinit
(
numpy
.
ones
(
1
)
*
5
)
n
=
astensor
(
numpy
.
ones
(
1
)
*
5
)
t
=
n
[
0
]
t
=
n
[
0
]
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
tval
=
eval_outputs
([
t
])
tval
=
eval_outputs
([
t
])
self
.
failUnless
(
tval
.
shape
==
())
self
.
failUnless
(
tval
.
shape
==
())
self
.
failUnless
(
tval
==
5.0
)
self
.
failUnless
(
tval
==
5.0
)
def
test1_ok_range_infinite
(
self
):
def
test1_ok_range_infinite
(
self
):
n
=
tinit
(
numpy
.
ones
(
3
)
*
5
)
n
=
astensor
(
numpy
.
ones
(
3
)
*
5
)
t
=
n
[
1
:]
t
=
n
[
1
:]
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
tval
=
eval_outputs
([
t
])
tval
=
eval_outputs
([
t
])
self
.
failUnless
(
tval
.
shape
==
(
2
,))
self
.
failUnless
(
tval
.
shape
==
(
2
,))
self
.
failUnless
(
tval
[
1
]
==
5.0
)
self
.
failUnless
(
tval
[
1
]
==
5.0
)
def
test1_ok_strided
(
self
):
def
test1_ok_strided
(
self
):
n
=
tinit
(
numpy
.
ones
(
5
)
*
5
)
n
=
astensor
(
numpy
.
ones
(
5
)
*
5
)
t
=
n
[
1
::
2
]
t
=
n
[
1
::
2
]
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
tval
=
eval_outputs
([
t
])
tval
=
eval_outputs
([
t
])
...
@@ -239,7 +239,7 @@ class T_subtensor(unittest.TestCase):
...
@@ -239,7 +239,7 @@ class T_subtensor(unittest.TestCase):
self
.
failUnless
(
tval
[
1
]
==
5.0
)
self
.
failUnless
(
tval
[
1
]
==
5.0
)
def
test2_err_bounds0
(
self
):
def
test2_err_bounds0
(
self
):
n
=
tinit
(
numpy
.
ones
((
2
,
3
))
*
5
)
n
=
astensor
(
numpy
.
ones
((
2
,
3
))
*
5
)
t
=
n
[
0
,
4
]
t
=
n
[
0
,
4
]
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
try
:
try
:
...
@@ -248,7 +248,7 @@ class T_subtensor(unittest.TestCase):
...
@@ -248,7 +248,7 @@ class T_subtensor(unittest.TestCase):
except
IndexError
,
e
:
except
IndexError
,
e
:
return
return
def
test2_err_bounds1
(
self
):
def
test2_err_bounds1
(
self
):
n
=
tinit
(
numpy
.
ones
((
2
,
3
))
*
5
)
n
=
astensor
(
numpy
.
ones
((
2
,
3
))
*
5
)
t
=
n
[
4
:
5
,
2
]
t
=
n
[
4
:
5
,
2
]
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
try
:
try
:
...
@@ -257,14 +257,14 @@ class T_subtensor(unittest.TestCase):
...
@@ -257,14 +257,14 @@ class T_subtensor(unittest.TestCase):
if
e
[
0
]
!=
'index out of bounds'
:
if
e
[
0
]
!=
'index out of bounds'
:
raise
raise
def
test2_ok_elem
(
self
):
def
test2_ok_elem
(
self
):
n
=
tinit
(
numpy
.
asarray
(
range
(
6
))
.
reshape
((
2
,
3
)))
n
=
astensor
(
numpy
.
asarray
(
range
(
6
))
.
reshape
((
2
,
3
)))
t
=
n
[
0
,
2
]
t
=
n
[
0
,
2
]
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
tval
=
eval_outputs
([
t
])
tval
=
eval_outputs
([
t
])
self
.
failUnless
(
tval
.
shape
==
())
self
.
failUnless
(
tval
.
shape
==
())
self
.
failUnless
(
numpy
.
all
(
tval
==
2
))
self
.
failUnless
(
numpy
.
all
(
tval
==
2
))
def
test2_ok_row
(
self
):
def
test2_ok_row
(
self
):
n
=
tinit
(
numpy
.
asarray
(
range
(
6
))
.
reshape
((
2
,
3
)))
n
=
astensor
(
numpy
.
asarray
(
range
(
6
))
.
reshape
((
2
,
3
)))
t
=
n
[
1
]
t
=
n
[
1
]
self
.
failIf
(
any
(
n
.
broadcastable
))
self
.
failIf
(
any
(
n
.
broadcastable
))
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
...
@@ -273,7 +273,7 @@ class T_subtensor(unittest.TestCase):
...
@@ -273,7 +273,7 @@ class T_subtensor(unittest.TestCase):
self
.
failUnless
(
numpy
.
all
(
tval
==
[
3
,
4
,
5
]))
self
.
failUnless
(
numpy
.
all
(
tval
==
[
3
,
4
,
5
]))
def
test2_ok_col
(
self
):
def
test2_ok_col
(
self
):
n
=
tinit
(
numpy
.
ones
((
2
,
3
))
*
5
)
n
=
astensor
(
numpy
.
ones
((
2
,
3
))
*
5
)
t
=
n
[:,
0
]
t
=
n
[:,
0
]
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
self
.
failIf
(
any
(
n
.
broadcastable
))
self
.
failIf
(
any
(
n
.
broadcastable
))
...
@@ -282,7 +282,7 @@ class T_subtensor(unittest.TestCase):
...
@@ -282,7 +282,7 @@ class T_subtensor(unittest.TestCase):
self
.
failUnless
(
numpy
.
all
(
tval
==
5.0
))
self
.
failUnless
(
numpy
.
all
(
tval
==
5.0
))
def
test2_ok_rows_finite
(
self
):
def
test2_ok_rows_finite
(
self
):
n
=
tinit
(
numpy
.
ones
((
4
,
3
))
*
5
)
n
=
astensor
(
numpy
.
ones
((
4
,
3
))
*
5
)
t
=
n
[
1
:
3
,
0
]
t
=
n
[
1
:
3
,
0
]
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
tval
=
eval_outputs
([
t
])
tval
=
eval_outputs
([
t
])
...
@@ -290,7 +290,7 @@ class T_subtensor(unittest.TestCase):
...
@@ -290,7 +290,7 @@ class T_subtensor(unittest.TestCase):
self
.
failUnless
(
numpy
.
all
(
tval
==
5.0
))
self
.
failUnless
(
numpy
.
all
(
tval
==
5.0
))
def
test2_ok_cols_infinite
(
self
):
def
test2_ok_cols_infinite
(
self
):
n
=
tinit
(
numpy
.
asarray
(
range
(
12
))
.
reshape
((
4
,
3
)))
n
=
astensor
(
numpy
.
asarray
(
range
(
12
))
.
reshape
((
4
,
3
)))
t
=
n
[
1
,
2
:]
t
=
n
[
1
,
2
:]
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
tval
=
eval_outputs
([
t
])
tval
=
eval_outputs
([
t
])
...
@@ -298,7 +298,7 @@ class T_subtensor(unittest.TestCase):
...
@@ -298,7 +298,7 @@ class T_subtensor(unittest.TestCase):
self
.
failUnless
(
numpy
.
all
(
tval
==
5
))
self
.
failUnless
(
numpy
.
all
(
tval
==
5
))
def
test2_ok_strided
(
self
):
def
test2_ok_strided
(
self
):
n
=
tinit
(
numpy
.
asarray
(
range
(
20
))
.
reshape
((
4
,
5
)))
n
=
astensor
(
numpy
.
asarray
(
range
(
20
))
.
reshape
((
4
,
5
)))
t
=
n
[
1
:
4
:
2
,
1
:
5
:
2
]
t
=
n
[
1
:
4
:
2
,
1
:
5
:
2
]
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
tval
=
eval_outputs
([
t
])
tval
=
eval_outputs
([
t
])
...
@@ -306,7 +306,7 @@ class T_subtensor(unittest.TestCase):
...
@@ -306,7 +306,7 @@ class T_subtensor(unittest.TestCase):
self
.
failUnless
(
numpy
.
all
(
tval
==
[[
6
,
8
],[
16
,
18
]]))
self
.
failUnless
(
numpy
.
all
(
tval
==
[[
6
,
8
],[
16
,
18
]]))
def
test3_ok_mat
(
self
):
def
test3_ok_mat
(
self
):
n
=
tinit
(
numpy
.
asarray
(
range
(
24
))
.
reshape
((
2
,
3
,
4
)))
n
=
astensor
(
numpy
.
asarray
(
range
(
24
))
.
reshape
((
2
,
3
,
4
)))
t
=
n
[
0
,
0
,
0
]
t
=
n
[
0
,
0
,
0
]
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
self
.
failUnless
(
t
.
owner
.
__class__
is
Subtensor
)
tval
=
eval_outputs
([
t
])
tval
=
eval_outputs
([
t
])
...
@@ -318,8 +318,8 @@ class T_add(unittest.TestCase):
...
@@ -318,8 +318,8 @@ class T_add(unittest.TestCase):
def
test_complex_all_ops
(
self
):
def
test_complex_all_ops
(
self
):
for
nbits
in
(
64
,
128
):
for
nbits
in
(
64
,
128
):
a
=
tinit
(
numpy
.
ones
(
3
,
dtype
=
'complex
%
i'
%
nbits
)
+
0.5
j
)
a
=
astensor
(
numpy
.
ones
(
3
,
dtype
=
'complex
%
i'
%
nbits
)
+
0.5
j
)
b
=
tinit
(
numpy
.
ones
(
3
,
dtype
=
'complex
%
i'
%
nbits
)
+
1.5
j
)
b
=
astensor
(
numpy
.
ones
(
3
,
dtype
=
'complex
%
i'
%
nbits
)
+
1.5
j
)
tests
=
((
"+"
,
lambda
x
,
y
:
x
+
y
),
tests
=
((
"+"
,
lambda
x
,
y
:
x
+
y
),
(
"-"
,
lambda
x
,
y
:
x
-
y
),
(
"-"
,
lambda
x
,
y
:
x
-
y
),
(
"*"
,
lambda
x
,
y
:
x
*
y
),
(
"*"
,
lambda
x
,
y
:
x
*
y
),
...
@@ -331,12 +331,12 @@ class T_add(unittest.TestCase):
...
@@ -331,12 +331,12 @@ class T_add(unittest.TestCase):
class
T_abs
(
unittest
.
TestCase
):
class
T_abs
(
unittest
.
TestCase
):
def
test_impl
(
self
):
def
test_impl
(
self
):
t
=
tinit
(
1.0
)
t
=
astensor
(
1.0
)
check_eq
(
self
,
t
,
abs
(
t
),
1.0
,
1.0
)
check_eq
(
self
,
t
,
abs
(
t
),
1.0
,
1.0
)
check_eq
(
self
,
t
,
abs
(
t
),
-
1.0
,
1.0
)
check_eq
(
self
,
t
,
abs
(
t
),
-
1.0
,
1.0
)
for
shape
in
(
2
,),
(
3
,
4
):
for
shape
in
(
2
,),
(
3
,
4
):
t
=
tinit
(
numpy
.
ones
(
shape
))
t
=
astensor
(
numpy
.
ones
(
shape
))
d
=
numpy
.
random
.
rand
(
*
shape
)
*
2
-
1.0
d
=
numpy
.
random
.
rand
(
*
shape
)
*
2
-
1.0
check_eq
(
self
,
t
,
abs
(
t
),
d
,
abs
(
d
))
check_eq
(
self
,
t
,
abs
(
t
),
d
,
abs
(
d
))
check_eq
(
self
,
t
,
abs
(
t
),
-
d
,
abs
(
-
d
))
check_eq
(
self
,
t
,
abs
(
t
),
-
d
,
abs
(
-
d
))
...
@@ -374,11 +374,11 @@ class T_fill(unittest.TestCase):
...
@@ -374,11 +374,11 @@ class T_fill(unittest.TestCase):
class
T_sum
(
unittest
.
TestCase
):
class
T_sum
(
unittest
.
TestCase
):
def
test_impl
(
self
):
def
test_impl
(
self
):
t
=
tinit
(
0.0
)
t
=
astensor
(
0.0
)
check_eq
(
self
,
t
,
Sum
(
t
)
.
out
,
1.0
,
1.0
)
check_eq
(
self
,
t
,
Sum
(
t
)
.
out
,
1.0
,
1.0
)
check_eq
(
self
,
t
,
Sum
(
t
)
.
out
,
-
1.0
,
-
1.0
)
check_eq
(
self
,
t
,
Sum
(
t
)
.
out
,
-
1.0
,
-
1.0
)
t
=
tinit
([
0.0
,
0.0
])
t
=
astensor
([
0.0
,
0.0
])
d
=
numpy
.
asarray
([
-
0.4
,
1.2
])
d
=
numpy
.
asarray
([
-
0.4
,
1.2
])
check_eq
(
self
,
t
,
Sum
(
t
)
.
out
,
d
,
numpy
.
sum
(
d
))
check_eq
(
self
,
t
,
Sum
(
t
)
.
out
,
d
,
numpy
.
sum
(
d
))
check_eq
(
self
,
t
,
Sum
(
t
)
.
out
,
-
d
,
-
numpy
.
sum
(
d
))
check_eq
(
self
,
t
,
Sum
(
t
)
.
out
,
-
d
,
-
numpy
.
sum
(
d
))
...
@@ -388,14 +388,14 @@ class T_mul(unittest.TestCase):
...
@@ -388,14 +388,14 @@ class T_mul(unittest.TestCase):
numpy
.
random
.
seed
([
1
,
2
,
3
,
4
])
numpy
.
random
.
seed
([
1
,
2
,
3
,
4
])
def
test_elemwise
(
self
):
def
test_elemwise
(
self
):
a
=
tinit
(
0.0
)
a
=
astensor
(
0.0
)
b
=
tinit
(
0.0
)
b
=
astensor
(
0.0
)
check_eq2
(
self
,
[
a
,
b
],
mul_elemwise
(
a
,
b
),
[
3.0
,
4.0
],
12.0
)
check_eq2
(
self
,
[
a
,
b
],
mul_elemwise
(
a
,
b
),
[
3.0
,
4.0
],
12.0
)
check_eq2
(
self
,
[
a
,
b
],
mul_elemwise
(
b
,
a
),
[
-
1.0
,
2.0
],
-
2.0
)
check_eq2
(
self
,
[
a
,
b
],
mul_elemwise
(
b
,
a
),
[
-
1.0
,
2.0
],
-
2.0
)
self
.
failUnless
(
isinstance
(
mul
(
a
,
b
)
.
owner
,
Scale
))
self
.
failUnless
(
isinstance
(
mul
(
a
,
b
)
.
owner
,
Scale
))
a
=
tinit
(
numpy
.
ones
(
2
))
a
=
astensor
(
numpy
.
ones
(
2
))
b
=
tinit
(
numpy
.
ones
(
2
))
b
=
astensor
(
numpy
.
ones
(
2
))
aa
=
numpy
.
asarray
([
-
0.5
,
4.0
])
aa
=
numpy
.
asarray
([
-
0.5
,
4.0
])
bb
=
numpy
.
asarray
([
-
0.5
,
2.0
])
bb
=
numpy
.
asarray
([
-
0.5
,
2.0
])
check_eq2
(
self
,
[
a
,
b
],
mul_elemwise
(
a
,
b
),
[
aa
,
bb
],
numpy
.
asarray
([
0.25
,
8.0
]))
check_eq2
(
self
,
[
a
,
b
],
mul_elemwise
(
a
,
b
),
[
aa
,
bb
],
numpy
.
asarray
([
0.25
,
8.0
]))
...
@@ -404,24 +404,24 @@ class T_mul(unittest.TestCase):
...
@@ -404,24 +404,24 @@ class T_mul(unittest.TestCase):
def
test_scalar
(
self
):
def
test_scalar
(
self
):
r
=
numpy
.
random
.
rand
(
2
,
3
)
r
=
numpy
.
random
.
rand
(
2
,
3
)
a
=
tinit
(
r
)
a
=
astensor
(
r
)
b
=
tinit
(
2.0
)
b
=
astensor
(
2.0
)
check_eq2
(
self
,
[
a
,
b
],
scale
(
a
,
b
),
[
r
,
2.0
],
r
*
2.0
)
check_eq2
(
self
,
[
a
,
b
],
scale
(
a
,
b
),
[
r
,
2.0
],
r
*
2.0
)
check_eq2
(
self
,
[
a
,
b
],
scale
(
a
,
b
),
[
r
,
4.0
],
r
*
4.0
)
check_eq2
(
self
,
[
a
,
b
],
scale
(
a
,
b
),
[
r
,
4.0
],
r
*
4.0
)
self
.
failUnless
(
b
.
data
==
2.0
)
self
.
failUnless
(
b
.
data
==
2.0
)
def
test_operator
(
self
):
def
test_operator
(
self
):
a
=
tinit
([
1
,
1
])
a
=
astensor
([
1
,
1
])
aa
=
tinit
([
1
,
1
])
aa
=
astensor
([
1
,
1
])
b
=
tinit
(
4
)
b
=
astensor
(
4
)
self
.
failUnless
(
isinstance
((
a
*
b
)
.
owner
,
Scale
))
self
.
failUnless
(
isinstance
((
a
*
b
)
.
owner
,
Scale
))
self
.
failUnless
(
isinstance
((
b
*
a
)
.
owner
,
Scale
))
self
.
failUnless
(
isinstance
((
b
*
a
)
.
owner
,
Scale
))
self
.
failUnless
(
isinstance
((
a
*
aa
)
.
owner
,
MulElemwise
))
self
.
failUnless
(
isinstance
((
a
*
aa
)
.
owner
,
MulElemwise
))
self
.
failUnless
(
isinstance
((
aa
*
a
)
.
owner
,
MulElemwise
))
self
.
failUnless
(
isinstance
((
aa
*
a
)
.
owner
,
MulElemwise
))
def
test_wrong_shapes
(
self
):
def
test_wrong_shapes
(
self
):
a
=
tinit
(
numpy
.
ones
(
3
))
a
=
astensor
(
numpy
.
ones
(
3
))
b
=
tinit
(
numpy
.
ones
(
4
))
b
=
astensor
(
numpy
.
ones
(
4
))
try
:
try
:
check_eq2
(
self
,
[
a
,
b
],
MulElemwise
(
a
,
b
)
.
out
,
check_eq2
(
self
,
[
a
,
b
],
MulElemwise
(
a
,
b
)
.
out
,
[
numpy
.
ones
(
3
),
numpy
.
ones
(
4
)],
1.0
)
[
numpy
.
ones
(
3
),
numpy
.
ones
(
4
)],
1.0
)
...
@@ -471,15 +471,14 @@ class _testCase_matinv(unittest.TestCase):
...
@@ -471,15 +471,14 @@ class _testCase_matinv(unittest.TestCase):
a
=
Tensor
(
'float64'
,
broadcastable
=
[
False
,
False
],
name
=
'a'
)
a
=
Tensor
(
'float64'
,
broadcastable
=
[
False
,
False
],
name
=
'a'
)
b
=
Tensor
(
'float64'
,
broadcastable
=
[
False
,
False
],
name
=
'b'
)
b
=
Tensor
(
'float64'
,
broadcastable
=
[
False
,
False
],
name
=
'b'
)
ab
=
a
*
b
ab
=
a
*
b
# Here, tinit actually uses the data allocated by numpy.
# Here, astensor actually uses the data allocated by numpy.
# TODO: Rename tinit to tensor
diff
=
ab
-
astensor
(
numpy
.
ones
((
dim
,
dim
)))
diff
=
ab
-
tinit
(
numpy
.
ones
((
dim
,
dim
)))
# Sum of squared errors
# Sum of squared errors
ssdiff
=
sum
((
diff
**
2.0
))
ssdiff
=
sum
((
diff
**
2.0
))
# May be able to abbreviate this by assuming default parameter
# May be able to abbreviate this by assuming default parameter
# TODO: Test that default works
# TODO: Test that default works
g_b
=
gradient
.
grad
(
ssdiff
,
b
,
tinit
(
numpy
.
ones
(
1
),
name
=
'g_cost'
))
g_b
=
gradient
.
grad
(
ssdiff
,
b
,
astensor
(
numpy
.
ones
(
1
),
name
=
'g_cost'
))
#g_b = gradient.grad(ssdiff, b) # This should be the abbreviated version
#g_b = gradient.grad(ssdiff, b) # This should be the abbreviated version
# compilation to function
# compilation to function
...
...
tensor.py
浏览文件 @
578eb363
...
@@ -67,7 +67,7 @@ class Tensor(BaseTensor):
...
@@ -67,7 +67,7 @@ class Tensor(BaseTensor):
def
__getslice__
(
self
,
*
args
):
return
subtensor
(
self
,
slice
(
*
args
))
def
__getslice__
(
self
,
*
args
):
return
subtensor
(
self
,
slice
(
*
args
))
# alternate Tensor constructor
# alternate Tensor constructor
def
tinit
(
data
,
broadcastable
=
None
,
role
=
None
,
name
=
None
):
def
astensor
(
data
,
broadcastable
=
None
,
role
=
None
,
name
=
None
):
"""Return a Tensor containing given data"""
"""Return a Tensor containing given data"""
data
=
numpy
.
asarray
(
data
)
data
=
numpy
.
asarray
(
data
)
if
broadcastable
is
None
:
if
broadcastable
is
None
:
...
@@ -90,7 +90,7 @@ def _scalar_switch(normal_f, scalar_f, scalar_f_reverse = None):
...
@@ -90,7 +90,7 @@ def _scalar_switch(normal_f, scalar_f, scalar_f_reverse = None):
if
isinstance
(
obj
,
Tensor
):
if
isinstance
(
obj
,
Tensor
):
return
obj
return
obj
else
:
else
:
return
tinit
(
obj
)
return
astensor
(
obj
)
x
,
y
=
as_tensor
(
x
),
as_tensor
(
y
)
x
,
y
=
as_tensor
(
x
),
as_tensor
(
y
)
if
0
not
in
y
.
broadcastable
:
if
0
not
in
y
.
broadcastable
:
return
scalar_f
(
x
,
y
)
return
scalar_f
(
x
,
y
)
...
@@ -119,7 +119,7 @@ def _as_tensor(obj):
...
@@ -119,7 +119,7 @@ def _as_tensor(obj):
if
isinstance
(
obj
,
Tensor
):
if
isinstance
(
obj
,
Tensor
):
return
obj
return
obj
else
:
else
:
return
tinit
(
obj
)
return
astensor
(
obj
)
class
_Op
(
BaseTensorOp
):
class
_Op
(
BaseTensorOp
):
"""A convenient base for the ops in this file"""
"""A convenient base for the ops in this file"""
...
...
编写
预览
Markdown
格式
0%
重试
或
添加新文件
添加附件
取消
您添加了
0
人
到此讨论。请谨慎行事。
请先完成此评论的编辑!
取消
请
注册
或者
登录
后发表评论