Skip to content
项目
群组
代码片段
帮助
当前项目
正在载入...
登录 / 注册
切换导航面板
P
pytensor
项目
项目
详情
活动
周期分析
仓库
仓库
文件
提交
分支
标签
贡献者
图表
比较
统计图
议题
0
议题
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
CI / CD
CI / CD
流水线
作业
日程
统计图
Wiki
Wiki
代码片段
代码片段
成员
成员
折叠边栏
关闭边栏
活动
图像
聊天
创建新问题
作业
提交
问题看板
Open sidebar
testgroup
pytensor
Commits
bcd5d52e
提交
bcd5d52e
authored
3月 24, 2017
作者:
amrithasuresh
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
1. Updated numpy as np
2. Fixed indentation
上级
f1bf4253
隐藏空白字符变更
内嵌
并排
正在显示
1 个修改的文件
包含
72 行增加
和
72 行删除
+72
-72
test_sharedvar.py
theano/tensor/tests/test_sharedvar.py
+72
-72
没有找到文件。
theano/tensor/tests/test_sharedvar.py
浏览文件 @
bcd5d52e
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
import
warnings
import
warnings
...
@@ -24,7 +24,7 @@ def makeSharedTester(shared_constructor_,
...
@@ -24,7 +24,7 @@ def makeSharedTester(shared_constructor_,
test_internal_type_
,
test_internal_type_
,
theano_fct_
,
theano_fct_
,
ref_fct_
,
ref_fct_
,
cast_value_
=
n
umpy
.
asarray
,
cast_value_
=
n
p
.
asarray
,
op_by_matrix_
=
False
,
op_by_matrix_
=
False
,
name
=
None
,
name
=
None
,
):
):
...
@@ -82,8 +82,8 @@ def makeSharedTester(shared_constructor_,
...
@@ -82,8 +82,8 @@ def makeSharedTester(shared_constructor_,
if
dtype
is
None
:
if
dtype
is
None
:
dtype
=
theano
.
config
.
floatX
dtype
=
theano
.
config
.
floatX
rng
=
n
umpy
.
random
.
RandomState
(
utt
.
fetch_seed
())
rng
=
n
p
.
random
.
RandomState
(
utt
.
fetch_seed
())
x
=
n
umpy
.
asarray
(
rng
.
uniform
(
0
,
1
,
[
2
,
4
]),
dtype
=
dtype
)
x
=
n
p
.
asarray
(
rng
.
uniform
(
0
,
1
,
[
2
,
4
]),
dtype
=
dtype
)
x
=
self
.
cast_value
(
x
)
x
=
self
.
cast_value
(
x
)
x_ref
=
self
.
ref_fct
(
x
)
x_ref
=
self
.
ref_fct
(
x
)
...
@@ -94,17 +94,17 @@ def makeSharedTester(shared_constructor_,
...
@@ -94,17 +94,17 @@ def makeSharedTester(shared_constructor_,
total_val
=
total_func
()
total_val
=
total_func
()
assert
n
umpy
.
allclose
(
self
.
ref_fct
(
x
),
total_val
)
assert
n
p
.
allclose
(
self
.
ref_fct
(
x
),
total_val
)
values_to_div
=
.
5
values_to_div
=
.
5
if
self
.
op_by_matrix
:
if
self
.
op_by_matrix
:
values_to_div
=
self
.
internal_type
(
n
umpy
.
ones
(
x
.
shape
,
dtype
=
dtype
)
/
2
)
# supported for cudandarray, but not ndarray.
values_to_div
=
self
.
internal_type
(
n
p
.
ones
(
x
.
shape
,
dtype
=
dtype
)
/
2
)
# supported for cudandarray, but not ndarray.
assert
self
.
test_internal_type
(
values_to_div
)
assert
self
.
test_internal_type
(
values_to_div
)
x
/=
values_to_div
x
/=
values_to_div
total_val_2
=
total_func
()
total_val_2
=
total_func
()
# value used to construct should not alias with internal
# value used to construct should not alias with internal
assert
n
umpy
.
allclose
(
total_val
,
total_val_2
)
assert
n
p
.
allclose
(
total_val
,
total_val_2
)
x
=
x_shared
.
get_value
(
borrow
=
False
)
x
=
x_shared
.
get_value
(
borrow
=
False
)
...
@@ -113,7 +113,7 @@ def makeSharedTester(shared_constructor_,
...
@@ -113,7 +113,7 @@ def makeSharedTester(shared_constructor_,
total_val_3
=
total_func
()
total_val_3
=
total_func
()
# value returned by access should not alias with internal
# value returned by access should not alias with internal
assert
n
umpy
.
allclose
(
total_val
,
total_val_3
)
assert
n
p
.
allclose
(
total_val
,
total_val_3
)
# in this case we can alias
# in this case we can alias
x
=
x_shared
.
get_value
(
borrow
=
True
)
x
=
x_shared
.
get_value
(
borrow
=
True
)
...
@@ -122,17 +122,17 @@ def makeSharedTester(shared_constructor_,
...
@@ -122,17 +122,17 @@ def makeSharedTester(shared_constructor_,
# this is not required by the contract but it is a feature we've
# this is not required by the contract but it is a feature we've
# implemented for some type of SharedVariable.
# implemented for some type of SharedVariable.
if
self
.
get_value_borrow_true_alias
:
if
self
.
get_value_borrow_true_alias
:
assert
n
umpy
.
allclose
(
self
.
ref_fct
(
x
),
total_func
())
assert
n
p
.
allclose
(
self
.
ref_fct
(
x
),
total_func
())
else
:
else
:
assert
n
umpy
.
allclose
(
x_ref
,
total_func
())
assert
n
p
.
allclose
(
x_ref
,
total_func
())
def
test_shape
(
self
):
def
test_shape
(
self
):
dtype
=
self
.
dtype
dtype
=
self
.
dtype
if
dtype
is
None
:
if
dtype
is
None
:
dtype
=
theano
.
config
.
floatX
dtype
=
theano
.
config
.
floatX
rng
=
n
umpy
.
random
.
RandomState
(
utt
.
fetch_seed
())
rng
=
n
p
.
random
.
RandomState
(
utt
.
fetch_seed
())
x
=
n
umpy
.
asarray
(
rng
.
uniform
(
0
,
1
,
[
2
,
4
]),
dtype
=
dtype
)
x
=
n
p
.
asarray
(
rng
.
uniform
(
0
,
1
,
[
2
,
4
]),
dtype
=
dtype
)
x
=
self
.
cast_value
(
x
)
x
=
self
.
cast_value
(
x
)
x_ref
=
self
.
ref_fct
(
x
)
x_ref
=
self
.
ref_fct
(
x
)
...
@@ -142,7 +142,7 @@ def makeSharedTester(shared_constructor_,
...
@@ -142,7 +142,7 @@ def makeSharedTester(shared_constructor_,
f
=
theano
.
function
([],
x_shared
.
shape
)
f
=
theano
.
function
([],
x_shared
.
shape
)
topo
=
f
.
maker
.
fgraph
.
toposort
()
topo
=
f
.
maker
.
fgraph
.
toposort
()
assert
n
umpy
.
all
(
f
()
==
(
2
,
4
))
assert
n
p
.
all
(
f
()
==
(
2
,
4
))
if
theano
.
config
.
mode
!=
'FAST_COMPILE'
:
if
theano
.
config
.
mode
!=
'FAST_COMPILE'
:
assert
len
(
topo
)
==
3
assert
len
(
topo
)
==
3
assert
isinstance
(
topo
[
0
]
.
op
,
tensor
.
opt
.
Shape_i
)
assert
isinstance
(
topo
[
0
]
.
op
,
tensor
.
opt
.
Shape_i
)
...
@@ -154,8 +154,8 @@ def makeSharedTester(shared_constructor_,
...
@@ -154,8 +154,8 @@ def makeSharedTester(shared_constructor_,
if
dtype
is
None
:
if
dtype
is
None
:
dtype
=
theano
.
config
.
floatX
dtype
=
theano
.
config
.
floatX
rng
=
n
umpy
.
random
.
RandomState
(
utt
.
fetch_seed
())
rng
=
n
p
.
random
.
RandomState
(
utt
.
fetch_seed
())
x
=
n
umpy
.
asarray
(
rng
.
uniform
(
0
,
1
,
[
2
,
4
]),
dtype
=
dtype
)
x
=
n
p
.
asarray
(
rng
.
uniform
(
0
,
1
,
[
2
,
4
]),
dtype
=
dtype
)
x
=
self
.
cast_value
(
x
)
x
=
self
.
cast_value
(
x
)
x_ref
=
self
.
ref_fct
(
x
)
x_ref
=
self
.
ref_fct
(
x
)
...
@@ -165,7 +165,7 @@ def makeSharedTester(shared_constructor_,
...
@@ -165,7 +165,7 @@ def makeSharedTester(shared_constructor_,
f
=
theano
.
function
([],
x_shared
.
shape
[
1
])
f
=
theano
.
function
([],
x_shared
.
shape
[
1
])
topo
=
f
.
maker
.
fgraph
.
toposort
()
topo
=
f
.
maker
.
fgraph
.
toposort
()
assert
n
umpy
.
all
(
f
()
==
(
4
))
assert
n
p
.
all
(
f
()
==
(
4
))
if
theano
.
config
.
mode
!=
'FAST_COMPILE'
:
if
theano
.
config
.
mode
!=
'FAST_COMPILE'
:
assert
len
(
topo
)
==
1
assert
len
(
topo
)
==
1
assert
isinstance
(
topo
[
0
]
.
op
,
tensor
.
opt
.
Shape_i
)
assert
isinstance
(
topo
[
0
]
.
op
,
tensor
.
opt
.
Shape_i
)
...
@@ -175,8 +175,8 @@ def makeSharedTester(shared_constructor_,
...
@@ -175,8 +175,8 @@ def makeSharedTester(shared_constructor_,
if
dtype
is
None
:
if
dtype
is
None
:
dtype
=
theano
.
config
.
floatX
dtype
=
theano
.
config
.
floatX
rng
=
n
umpy
.
random
.
RandomState
(
utt
.
fetch_seed
())
rng
=
n
p
.
random
.
RandomState
(
utt
.
fetch_seed
())
x
=
n
umpy
.
asarray
(
rng
.
uniform
(
0
,
1
,
[
2
,
4
]),
dtype
=
dtype
)
x
=
n
p
.
asarray
(
rng
.
uniform
(
0
,
1
,
[
2
,
4
]),
dtype
=
dtype
)
x
=
self
.
cast_value
(
x
)
x
=
self
.
cast_value
(
x
)
x_ref
=
self
.
ref_fct
(
x
)
x_ref
=
self
.
ref_fct
(
x
)
...
@@ -193,12 +193,12 @@ def makeSharedTester(shared_constructor_,
...
@@ -193,12 +193,12 @@ def makeSharedTester(shared_constructor_,
if
self
.
op_by_matrix
:
if
self
.
op_by_matrix
:
# supported for cudandarray, but not ndarray.
# supported for cudandarray, but not ndarray.
values_to_div
=
self
.
internal_type
(
values_to_div
=
self
.
internal_type
(
n
umpy
.
ones
(
x
.
shape
,
dtype
=
dtype
)
/
2
)
n
p
.
ones
(
x
.
shape
,
dtype
=
dtype
)
/
2
)
x
/=
values_to_div
# supported by ndarray and CudaNdarray
x
/=
values_to_div
# supported by ndarray and CudaNdarray
# this is not required by the contract but it is a feature we can
# this is not required by the contract but it is a feature we can
# implement for some type of SharedVariable.
# implement for some type of SharedVariable.
assert
n
umpy
.
allclose
(
self
.
ref_fct
(
x
),
total_func
())
assert
n
p
.
allclose
(
self
.
ref_fct
(
x
),
total_func
())
x
=
x_shared
.
get_value
(
borrow
=
False
,
return_internal_type
=
True
)
x
=
x_shared
.
get_value
(
borrow
=
False
,
return_internal_type
=
True
)
assert
self
.
test_internal_type
(
x
)
assert
self
.
test_internal_type
(
x
)
...
@@ -206,7 +206,7 @@ def makeSharedTester(shared_constructor_,
...
@@ -206,7 +206,7 @@ def makeSharedTester(shared_constructor_,
x
/=
values_to_div
# supported by ndarray and CudaNdarray
x
/=
values_to_div
# supported by ndarray and CudaNdarray
# this is required by the contract
# this is required by the contract
assert
not
n
umpy
.
allclose
(
self
.
ref_fct
(
x
),
total_func
())
assert
not
n
p
.
allclose
(
self
.
ref_fct
(
x
),
total_func
())
def
test_get_value
(
self
):
def
test_get_value
(
self
):
"""
"""
...
@@ -216,8 +216,8 @@ def makeSharedTester(shared_constructor_,
...
@@ -216,8 +216,8 @@ def makeSharedTester(shared_constructor_,
if
dtype
is
None
:
if
dtype
is
None
:
dtype
=
theano
.
config
.
floatX
dtype
=
theano
.
config
.
floatX
rng
=
n
umpy
.
random
.
RandomState
(
utt
.
fetch_seed
())
rng
=
n
p
.
random
.
RandomState
(
utt
.
fetch_seed
())
x_orig
=
n
umpy
.
asarray
(
rng
.
uniform
(
0
,
1
,
[
2
,
4
]),
dtype
=
dtype
)
x_orig
=
n
p
.
asarray
(
rng
.
uniform
(
0
,
1
,
[
2
,
4
]),
dtype
=
dtype
)
x_cast
=
self
.
cast_value
(
x_orig
)
x_cast
=
self
.
cast_value
(
x_orig
)
if
self
.
shared_constructor_accept_ndarray
:
if
self
.
shared_constructor_accept_ndarray
:
x_shared
=
self
.
shared_constructor
(
x_orig
,
borrow
=
False
)
x_shared
=
self
.
shared_constructor
(
x_orig
,
borrow
=
False
)
...
@@ -231,8 +231,8 @@ def makeSharedTester(shared_constructor_,
...
@@ -231,8 +231,8 @@ def makeSharedTester(shared_constructor_,
if
dtype
is
None
:
if
dtype
is
None
:
dtype
=
theano
.
config
.
floatX
dtype
=
theano
.
config
.
floatX
rng
=
n
umpy
.
random
.
RandomState
(
utt
.
fetch_seed
())
rng
=
n
p
.
random
.
RandomState
(
utt
.
fetch_seed
())
x
=
n
umpy
.
asarray
(
rng
.
uniform
(
0
,
1
,
[
2
,
4
]),
dtype
=
dtype
)
x
=
n
p
.
asarray
(
rng
.
uniform
(
0
,
1
,
[
2
,
4
]),
dtype
=
dtype
)
x
=
self
.
cast_value
(
x
)
x
=
self
.
cast_value
(
x
)
x_orig
=
x
x_orig
=
x
...
@@ -247,7 +247,7 @@ def makeSharedTester(shared_constructor_,
...
@@ -247,7 +247,7 @@ def makeSharedTester(shared_constructor_,
values_to_div
=
.
5
values_to_div
=
.
5
if
self
.
op_by_matrix
:
if
self
.
op_by_matrix
:
# supported for cudandarray, but not ndarray.
# supported for cudandarray, but not ndarray.
values_to_div
=
self
.
internal_type
(
n
umpy
.
ones
(
x
.
shape
,
dtype
=
dtype
)
/
2
)
values_to_div
=
self
.
internal_type
(
n
p
.
ones
(
x
.
shape
,
dtype
=
dtype
)
/
2
)
assert
self
.
test_internal_type
(
values_to_div
)
assert
self
.
test_internal_type
(
values_to_div
)
# test if that theano shared variable optimize set_value(borrow=True)
# test if that theano shared variable optimize set_value(borrow=True)
...
@@ -260,7 +260,7 @@ def makeSharedTester(shared_constructor_,
...
@@ -260,7 +260,7 @@ def makeSharedTester(shared_constructor_,
assert
x
is
get_x
assert
x
is
get_x
else
:
else
:
assert
x
is
not
get_x
assert
x
is
not
get_x
assert
n
umpy
.
allclose
(
self
.
ref_fct
(
numpy
.
asarray
(
x_orig
)
/.
5
),
self
.
ref_fct
(
x
))
assert
n
p
.
allclose
(
self
.
ref_fct
(
np
.
asarray
(
x_orig
)
/.
5
),
self
.
ref_fct
(
x
))
# test optimized get set value on the gpu(don't pass data to the cpu)
# test optimized get set value on the gpu(don't pass data to the cpu)
get_x
=
x_shared
.
get_value
(
borrow
=
True
,
return_internal_type
=
True
)
get_x
=
x_shared
.
get_value
(
borrow
=
True
,
return_internal_type
=
True
)
...
@@ -280,8 +280,8 @@ def makeSharedTester(shared_constructor_,
...
@@ -280,8 +280,8 @@ def makeSharedTester(shared_constructor_,
if
dtype
is
None
:
if
dtype
is
None
:
dtype
=
theano
.
config
.
floatX
dtype
=
theano
.
config
.
floatX
rng
=
n
umpy
.
random
.
RandomState
(
utt
.
fetch_seed
())
rng
=
n
p
.
random
.
RandomState
(
utt
.
fetch_seed
())
x
=
n
umpy
.
asarray
(
rng
.
uniform
(
1
,
2
,
[
4
,
2
]),
dtype
=
dtype
)
x
=
n
p
.
asarray
(
rng
.
uniform
(
1
,
2
,
[
4
,
2
]),
dtype
=
dtype
)
x
=
self
.
cast_value
(
x
)
x
=
self
.
cast_value
(
x
)
x_ref
=
self
.
ref_fct
(
x
)
x_ref
=
self
.
ref_fct
(
x
)
...
@@ -293,20 +293,20 @@ def makeSharedTester(shared_constructor_,
...
@@ -293,20 +293,20 @@ def makeSharedTester(shared_constructor_,
total_val
=
total_func
()
total_val
=
total_func
()
assert
n
umpy
.
allclose
(
self
.
ref_fct
(
x
),
total_val
)
assert
n
p
.
allclose
(
self
.
ref_fct
(
x
),
total_val
)
values_to_div
=
.
5
values_to_div
=
.
5
if
self
.
op_by_matrix
:
if
self
.
op_by_matrix
:
# supported for cudandarray, but not ndarray.
# supported for cudandarray, but not ndarray.
values_to_div
=
self
.
internal_type
(
n
umpy
.
ones
(
x
.
shape
,
dtype
=
dtype
)
/
2
)
values_to_div
=
self
.
internal_type
(
n
p
.
ones
(
x
.
shape
,
dtype
=
dtype
)
/
2
)
assert
self
.
test_internal_type
(
values_to_div
)
assert
self
.
test_internal_type
(
values_to_div
)
x
/=
values_to_div
x
/=
values_to_div
# not required by the contract but it is a feature we've implemented
# not required by the contract but it is a feature we've implemented
if
self
.
shared_borrow_true_alias
:
if
self
.
shared_borrow_true_alias
:
assert
n
umpy
.
allclose
(
self
.
ref_fct
(
x
),
total_func
())
assert
n
p
.
allclose
(
self
.
ref_fct
(
x
),
total_func
())
else
:
else
:
assert
n
umpy
.
allclose
(
x_ref
,
total_func
())
assert
n
p
.
allclose
(
x_ref
,
total_func
())
def
test_inplace_set_value
(
self
):
def
test_inplace_set_value
(
self
):
"""
"""
...
@@ -319,25 +319,25 @@ def makeSharedTester(shared_constructor_,
...
@@ -319,25 +319,25 @@ def makeSharedTester(shared_constructor_,
shp
=
(
100
//
4
,
1024
)
# 100KB
shp
=
(
100
//
4
,
1024
)
# 100KB
x
=
n
umpy
.
zeros
(
shp
,
dtype
=
dtype
)
x
=
n
p
.
zeros
(
shp
,
dtype
=
dtype
)
x
=
self
.
cast_value
(
x
)
x
=
self
.
cast_value
(
x
)
x_shared
=
self
.
shared_constructor
(
x
,
borrow
=
True
)
x_shared
=
self
.
shared_constructor
(
x
,
borrow
=
True
)
old_data
=
x_shared
.
container
.
storage
[
0
]
old_data
=
x_shared
.
container
.
storage
[
0
]
nd
=
n
umpy
.
ones
(
shp
,
dtype
=
dtype
)
nd
=
n
p
.
ones
(
shp
,
dtype
=
dtype
)
if
x
.
__class__
.
__name__
!=
'csr_matrix'
:
if
x
.
__class__
.
__name__
!=
'csr_matrix'
:
# sparse matrix don't support inplace affectation
# sparse matrix don't support inplace affectation
x_shared
.
container
.
value
[:]
=
nd
x_shared
.
container
.
value
[:]
=
nd
assert
(
n
umpy
.
asarray
(
x_shared
.
get_value
(
borrow
=
True
))
==
nd
)
.
all
()
assert
(
n
p
.
asarray
(
x_shared
.
get_value
(
borrow
=
True
))
==
nd
)
.
all
()
# This should always share value!
# This should always share value!
assert
may_share_memory
(
old_data
,
x_shared
.
container
.
storage
[
0
])
assert
may_share_memory
(
old_data
,
x_shared
.
container
.
storage
[
0
])
assert
may_share_memory
(
old_data
,
x_shared
.
get_value
(
borrow
=
True
,
return_internal_type
=
True
))
assert
may_share_memory
(
old_data
,
x_shared
.
get_value
(
borrow
=
True
,
return_internal_type
=
True
))
nd
[
0
]
+=
1
nd
[
0
]
+=
1
x_shared
.
container
.
value
[
0
]
=
nd
[
0
]
x_shared
.
container
.
value
[
0
]
=
nd
[
0
]
assert
(
n
umpy
.
asarray
(
x_shared
.
get_value
(
borrow
=
True
)[
0
])
==
nd
[
0
])
.
all
()
assert
(
n
p
.
asarray
(
x_shared
.
get_value
(
borrow
=
True
)[
0
])
==
nd
[
0
])
.
all
()
assert
(
n
umpy
.
asarray
(
x_shared
.
get_value
(
borrow
=
True
)[
1
:])
==
nd
[
1
:])
.
all
()
assert
(
n
p
.
asarray
(
x_shared
.
get_value
(
borrow
=
True
)[
1
:])
==
nd
[
1
:])
.
all
()
# This should always share value!
# This should always share value!
assert
may_share_memory
(
old_data
,
x_shared
.
container
.
storage
[
0
])
assert
may_share_memory
(
old_data
,
x_shared
.
container
.
storage
[
0
])
assert
may_share_memory
(
old_data
,
x_shared
.
get_value
(
borrow
=
True
,
return_internal_type
=
True
))
assert
may_share_memory
(
old_data
,
x_shared
.
get_value
(
borrow
=
True
,
return_internal_type
=
True
))
...
@@ -347,7 +347,7 @@ def makeSharedTester(shared_constructor_,
...
@@ -347,7 +347,7 @@ def makeSharedTester(shared_constructor_,
nd
+=
1
nd
+=
1
# THIS DON't DO WHAT WE EXPECT the contain of a is not updated for CudaNdarray, but it is for ndarray
# THIS DON't DO WHAT WE EXPECT the contain of a is not updated for CudaNdarray, but it is for ndarray
x_shared
.
get_value
(
borrow
=
True
)[:]
=
nd
x_shared
.
get_value
(
borrow
=
True
)[:]
=
nd
#assert (n
umpy
.asarray(x_shared.get_value(borrow=True))!=nd).all()
#assert (n
p
.asarray(x_shared.get_value(borrow=True))!=nd).all()
assert
may_share_memory
(
old_data
,
x_shared
.
container
.
storage
[
0
])
assert
may_share_memory
(
old_data
,
x_shared
.
container
.
storage
[
0
])
x_shared
.
get_value
(
borrow
=
True
)
x_shared
.
get_value
(
borrow
=
True
)
...
@@ -355,7 +355,7 @@ def makeSharedTester(shared_constructor_,
...
@@ -355,7 +355,7 @@ def makeSharedTester(shared_constructor_,
nd
+=
1
nd
+=
1
old_data
=
x_shared
.
container
.
storage
[
0
]
old_data
=
x_shared
.
container
.
storage
[
0
]
x_shared
.
set_value
(
nd
,
borrow
=
False
)
x_shared
.
set_value
(
nd
,
borrow
=
False
)
assert
n
umpy
.
allclose
(
self
.
ref_fct
(
x_shared
.
get_value
(
borrow
=
True
)),
assert
n
p
.
allclose
(
self
.
ref_fct
(
x_shared
.
get_value
(
borrow
=
True
)),
self
.
ref_fct
(
self
.
cast_value
(
nd
)))
self
.
ref_fct
(
self
.
cast_value
(
nd
)))
assert
may_share_memory
(
old_data
,
x_shared
.
container
.
storage
[
0
])
==
self
.
set_value_inplace
assert
may_share_memory
(
old_data
,
x_shared
.
container
.
storage
[
0
])
==
self
.
set_value_inplace
...
@@ -364,7 +364,7 @@ def makeSharedTester(shared_constructor_,
...
@@ -364,7 +364,7 @@ def makeSharedTester(shared_constructor_,
nd
+=
1
nd
+=
1
old_data
=
x_shared
.
container
.
storage
[
0
]
old_data
=
x_shared
.
container
.
storage
[
0
]
x_shared
.
set_value
(
self
.
cast_value
(
nd
),
borrow
=
False
)
x_shared
.
set_value
(
self
.
cast_value
(
nd
),
borrow
=
False
)
assert
n
umpy
.
allclose
(
self
.
ref_fct
(
x_shared
.
get_value
(
borrow
=
True
)),
assert
n
p
.
allclose
(
self
.
ref_fct
(
x_shared
.
get_value
(
borrow
=
True
)),
self
.
ref_fct
(
self
.
cast_value
(
nd
)))
self
.
ref_fct
(
self
.
cast_value
(
nd
)))
assert
may_share_memory
(
old_data
,
x_shared
.
container
.
storage
[
0
])
==
self
.
set_cast_value_inplace
assert
may_share_memory
(
old_data
,
x_shared
.
container
.
storage
[
0
])
==
self
.
set_cast_value_inplace
...
@@ -372,7 +372,7 @@ def makeSharedTester(shared_constructor_,
...
@@ -372,7 +372,7 @@ def makeSharedTester(shared_constructor_,
nd
+=
1
nd
+=
1
old_data
=
x_shared
.
container
.
storage
[
0
]
old_data
=
x_shared
.
container
.
storage
[
0
]
x_shared
.
set_value
(
nd
.
copy
(),
borrow
=
True
)
x_shared
.
set_value
(
nd
.
copy
(),
borrow
=
True
)
assert
n
umpy
.
allclose
(
self
.
ref_fct
(
x_shared
.
get_value
(
borrow
=
True
)),
assert
n
p
.
allclose
(
self
.
ref_fct
(
x_shared
.
get_value
(
borrow
=
True
)),
self
.
ref_fct
(
self
.
cast_value
(
nd
)))
self
.
ref_fct
(
self
.
cast_value
(
nd
)))
assert
may_share_memory
(
old_data
,
x_shared
.
container
.
storage
[
0
])
==
self
.
set_value_inplace
assert
may_share_memory
(
old_data
,
x_shared
.
container
.
storage
[
0
])
==
self
.
set_value_inplace
...
@@ -380,7 +380,7 @@ def makeSharedTester(shared_constructor_,
...
@@ -380,7 +380,7 @@ def makeSharedTester(shared_constructor_,
nd
+=
1
nd
+=
1
old_data
=
x_shared
.
container
.
storage
[
0
]
old_data
=
x_shared
.
container
.
storage
[
0
]
x_shared
.
set_value
(
self
.
cast_value
(
nd
.
copy
()),
borrow
=
True
)
x_shared
.
set_value
(
self
.
cast_value
(
nd
.
copy
()),
borrow
=
True
)
assert
n
umpy
.
allclose
(
self
.
ref_fct
(
x_shared
.
get_value
(
borrow
=
True
)),
self
.
ref_fct
(
self
.
cast_value
(
nd
)))
assert
n
p
.
allclose
(
self
.
ref_fct
(
x_shared
.
get_value
(
borrow
=
True
)),
self
.
ref_fct
(
self
.
cast_value
(
nd
)))
assert
may_share_memory
(
old_data
,
x_shared
.
container
.
storage
[
0
])
==
self
.
set_cast_value_inplace
assert
may_share_memory
(
old_data
,
x_shared
.
container
.
storage
[
0
])
==
self
.
set_cast_value_inplace
def
test_specify_shape
(
self
):
def
test_specify_shape
(
self
):
...
@@ -388,19 +388,19 @@ def makeSharedTester(shared_constructor_,
...
@@ -388,19 +388,19 @@ def makeSharedTester(shared_constructor_,
if
dtype
is
None
:
if
dtype
is
None
:
dtype
=
theano
.
config
.
floatX
dtype
=
theano
.
config
.
floatX
rng
=
n
umpy
.
random
.
RandomState
(
utt
.
fetch_seed
())
rng
=
n
p
.
random
.
RandomState
(
utt
.
fetch_seed
())
x1_1
=
n
umpy
.
asarray
(
rng
.
uniform
(
1
,
2
,
[
4
,
2
]),
dtype
=
dtype
)
x1_1
=
n
p
.
asarray
(
rng
.
uniform
(
1
,
2
,
[
4
,
2
]),
dtype
=
dtype
)
x1_1
=
self
.
cast_value
(
x1_1
)
x1_1
=
self
.
cast_value
(
x1_1
)
x1_2
=
n
umpy
.
asarray
(
rng
.
uniform
(
1
,
2
,
[
4
,
2
]),
dtype
=
dtype
)
x1_2
=
n
p
.
asarray
(
rng
.
uniform
(
1
,
2
,
[
4
,
2
]),
dtype
=
dtype
)
x1_2
=
self
.
cast_value
(
x1_2
)
x1_2
=
self
.
cast_value
(
x1_2
)
x2
=
n
umpy
.
asarray
(
rng
.
uniform
(
1
,
2
,
[
4
,
3
]),
dtype
=
dtype
)
x2
=
n
p
.
asarray
(
rng
.
uniform
(
1
,
2
,
[
4
,
3
]),
dtype
=
dtype
)
x2
=
self
.
cast_value
(
x2
)
x2
=
self
.
cast_value
(
x2
)
# Test that we can replace with values of the same shape
# Test that we can replace with values of the same shape
x1_shared
=
self
.
shared_constructor
(
x1_1
)
x1_shared
=
self
.
shared_constructor
(
x1_1
)
x1_specify_shape
=
tensor
.
specify_shape
(
x1_shared
,
x1_1
.
shape
)
x1_specify_shape
=
tensor
.
specify_shape
(
x1_shared
,
x1_1
.
shape
)
x1_shared
.
set_value
(
x1_2
)
x1_shared
.
set_value
(
x1_2
)
assert
n
umpy
.
allclose
(
self
.
ref_fct
(
x1_shared
.
get_value
(
borrow
=
True
)),
assert
n
p
.
allclose
(
self
.
ref_fct
(
x1_shared
.
get_value
(
borrow
=
True
)),
self
.
ref_fct
(
x1_2
))
self
.
ref_fct
(
x1_2
))
shape_op_fct
=
theano
.
function
([],
x1_shared
.
shape
)
shape_op_fct
=
theano
.
function
([],
x1_shared
.
shape
)
topo
=
shape_op_fct
.
maker
.
fgraph
.
toposort
()
topo
=
shape_op_fct
.
maker
.
fgraph
.
toposort
()
...
@@ -412,14 +412,14 @@ def makeSharedTester(shared_constructor_,
...
@@ -412,14 +412,14 @@ def makeSharedTester(shared_constructor_,
# Test that we forward the input
# Test that we forward the input
specify_shape_fct
=
theano
.
function
([],
x1_specify_shape
)
specify_shape_fct
=
theano
.
function
([],
x1_specify_shape
)
assert
n
umpy
.
all
(
self
.
ref_fct
(
specify_shape_fct
())
==
assert
n
p
.
all
(
self
.
ref_fct
(
specify_shape_fct
())
==
self
.
ref_fct
(
x1_2
))
self
.
ref_fct
(
x1_2
))
topo_specify
=
specify_shape_fct
.
maker
.
fgraph
.
toposort
()
topo_specify
=
specify_shape_fct
.
maker
.
fgraph
.
toposort
()
assert
len
(
topo_specify
)
==
2
assert
len
(
topo_specify
)
==
2
# Test that we put the shape info into the graph
# Test that we put the shape info into the graph
shape_constant_fct
=
theano
.
function
([],
x1_specify_shape
.
shape
)
shape_constant_fct
=
theano
.
function
([],
x1_specify_shape
.
shape
)
assert
n
umpy
.
all
(
shape_constant_fct
()
==
shape_op_fct
())
assert
n
p
.
all
(
shape_constant_fct
()
==
shape_op_fct
())
topo_cst
=
shape_constant_fct
.
maker
.
fgraph
.
toposort
()
topo_cst
=
shape_constant_fct
.
maker
.
fgraph
.
toposort
()
if
theano
.
config
.
mode
!=
'FAST_COMPILE'
:
if
theano
.
config
.
mode
!=
'FAST_COMPILE'
:
assert
len
(
topo_cst
)
==
1
assert
len
(
topo_cst
)
==
1
...
@@ -454,12 +454,12 @@ def makeSharedTester(shared_constructor_,
...
@@ -454,12 +454,12 @@ def makeSharedTester(shared_constructor_,
if
dtype
is
None
:
if
dtype
is
None
:
dtype
=
theano
.
config
.
floatX
dtype
=
theano
.
config
.
floatX
rng
=
n
umpy
.
random
.
RandomState
(
utt
.
fetch_seed
())
rng
=
n
p
.
random
.
RandomState
(
utt
.
fetch_seed
())
x1_1
=
n
umpy
.
asarray
(
rng
.
uniform
(
1
,
2
,
[
4
,
2
]),
dtype
=
dtype
)
x1_1
=
n
p
.
asarray
(
rng
.
uniform
(
1
,
2
,
[
4
,
2
]),
dtype
=
dtype
)
x1_1
=
self
.
cast_value
(
x1_1
)
x1_1
=
self
.
cast_value
(
x1_1
)
x1_2
=
n
umpy
.
asarray
(
rng
.
uniform
(
1
,
2
,
[
4
,
2
]),
dtype
=
dtype
)
x1_2
=
n
p
.
asarray
(
rng
.
uniform
(
1
,
2
,
[
4
,
2
]),
dtype
=
dtype
)
x1_2
=
self
.
cast_value
(
x1_2
)
x1_2
=
self
.
cast_value
(
x1_2
)
x2
=
n
umpy
.
asarray
(
rng
.
uniform
(
1
,
2
,
[
5
,
2
]),
dtype
=
dtype
)
x2
=
n
p
.
asarray
(
rng
.
uniform
(
1
,
2
,
[
5
,
2
]),
dtype
=
dtype
)
x2
=
self
.
cast_value
(
x2
)
x2
=
self
.
cast_value
(
x2
)
# Test that we can replace with values of the same shape
# Test that we can replace with values of the same shape
...
@@ -468,7 +468,7 @@ def makeSharedTester(shared_constructor_,
...
@@ -468,7 +468,7 @@ def makeSharedTester(shared_constructor_,
(
tensor
.
as_tensor_variable
(
x1_1
.
shape
[
0
]),
(
tensor
.
as_tensor_variable
(
x1_1
.
shape
[
0
]),
x1_shared
.
shape
[
1
]))
x1_shared
.
shape
[
1
]))
x1_shared
.
set_value
(
x1_2
)
x1_shared
.
set_value
(
x1_2
)
assert
n
umpy
.
allclose
(
assert
n
p
.
allclose
(
self
.
ref_fct
(
x1_shared
.
get_value
(
borrow
=
True
)),
self
.
ref_fct
(
x1_shared
.
get_value
(
borrow
=
True
)),
self
.
ref_fct
(
x1_2
))
self
.
ref_fct
(
x1_2
))
shape_op_fct
=
theano
.
function
([],
x1_shared
.
shape
)
shape_op_fct
=
theano
.
function
([],
x1_shared
.
shape
)
...
@@ -484,7 +484,7 @@ def makeSharedTester(shared_constructor_,
...
@@ -484,7 +484,7 @@ def makeSharedTester(shared_constructor_,
specify_shape_fct
=
theano
.
function
([],
x1_specify_shape
)
specify_shape_fct
=
theano
.
function
([],
x1_specify_shape
)
specify_shape_fct
()
specify_shape_fct
()
# theano.printing.debugprint(specify_shape_fct)
# theano.printing.debugprint(specify_shape_fct)
assert
n
umpy
.
all
(
self
.
ref_fct
(
specify_shape_fct
())
assert
n
p
.
all
(
self
.
ref_fct
(
specify_shape_fct
())
==
self
.
ref_fct
(
x1_2
))
==
self
.
ref_fct
(
x1_2
))
topo_specify
=
specify_shape_fct
.
maker
.
fgraph
.
toposort
()
topo_specify
=
specify_shape_fct
.
maker
.
fgraph
.
toposort
()
if
theano
.
config
.
mode
!=
'FAST_COMPILE'
:
if
theano
.
config
.
mode
!=
'FAST_COMPILE'
:
...
@@ -493,7 +493,7 @@ def makeSharedTester(shared_constructor_,
...
@@ -493,7 +493,7 @@ def makeSharedTester(shared_constructor_,
# Test that we put the shape info into the graph
# Test that we put the shape info into the graph
shape_constant_fct
=
theano
.
function
([],
x1_specify_shape
.
shape
)
shape_constant_fct
=
theano
.
function
([],
x1_specify_shape
.
shape
)
# theano.printing.debugprint(shape_constant_fct)
# theano.printing.debugprint(shape_constant_fct)
assert
n
umpy
.
all
(
shape_constant_fct
()
==
shape_op_fct
())
assert
n
p
.
all
(
shape_constant_fct
()
==
shape_op_fct
())
topo_cst
=
shape_constant_fct
.
maker
.
fgraph
.
toposort
()
topo_cst
=
shape_constant_fct
.
maker
.
fgraph
.
toposort
()
if
theano
.
config
.
mode
!=
'FAST_COMPILE'
:
if
theano
.
config
.
mode
!=
'FAST_COMPILE'
:
assert
len
(
topo_cst
)
==
2
assert
len
(
topo_cst
)
==
2
...
@@ -516,14 +516,14 @@ def makeSharedTester(shared_constructor_,
...
@@ -516,14 +516,14 @@ def makeSharedTester(shared_constructor_,
if
dtype
is
None
:
if
dtype
is
None
:
dtype
=
theano
.
config
.
floatX
dtype
=
theano
.
config
.
floatX
rng
=
n
umpy
.
random
.
RandomState
(
utt
.
fetch_seed
())
rng
=
n
p
.
random
.
RandomState
(
utt
.
fetch_seed
())
a
=
n
umpy
.
asarray
(
rng
.
uniform
(
1
,
2
,
[
40
,
40
]),
dtype
=
dtype
)
a
=
n
p
.
asarray
(
rng
.
uniform
(
1
,
2
,
[
40
,
40
]),
dtype
=
dtype
)
a
=
self
.
cast_value
(
a
)
a
=
self
.
cast_value
(
a
)
a_shared
=
self
.
shared_constructor
(
a
)
a_shared
=
self
.
shared_constructor
(
a
)
b
=
n
umpy
.
asarray
(
rng
.
uniform
(
1
,
2
,
[
40
,
40
]),
dtype
=
dtype
)
b
=
n
p
.
asarray
(
rng
.
uniform
(
1
,
2
,
[
40
,
40
]),
dtype
=
dtype
)
b
=
self
.
cast_value
(
b
)
b
=
self
.
cast_value
(
b
)
b_shared
=
self
.
shared_constructor
(
b
)
b_shared
=
self
.
shared_constructor
(
b
)
s
=
n
umpy
.
zeros
((
40
,
40
),
dtype
=
dtype
)
s
=
n
p
.
zeros
((
40
,
40
),
dtype
=
dtype
)
s
=
self
.
cast_value
(
s
)
s
=
self
.
cast_value
(
s
)
s_shared
=
self
.
shared_constructor
(
s
)
s_shared
=
self
.
shared_constructor
(
s
)
f
=
theano
.
function
([],
f
=
theano
.
function
([],
...
@@ -546,7 +546,7 @@ def makeSharedTester(shared_constructor_,
...
@@ -546,7 +546,7 @@ def makeSharedTester(shared_constructor_,
+
s_shared_specify
)])
+
s_shared_specify
)])
topo
=
f
.
maker
.
fgraph
.
toposort
()
topo
=
f
.
maker
.
fgraph
.
toposort
()
shp
=
f
()
shp
=
f
()
assert
n
umpy
.
all
(
shp
==
(
40
,
40
))
assert
n
p
.
all
(
shp
==
(
40
,
40
))
if
theano
.
config
.
mode
!=
'FAST_COMPILE'
:
if
theano
.
config
.
mode
!=
'FAST_COMPILE'
:
assert
sum
([
node
.
op
.
__class__
.
__name__
in
[
"Gemm"
,
"GpuGemm"
,
"StructuredDot"
]
for
node
in
topo
])
==
1
assert
sum
([
node
.
op
.
__class__
.
__name__
in
[
"Gemm"
,
"GpuGemm"
,
"StructuredDot"
]
for
node
in
topo
])
==
1
assert
all
(
node
.
op
==
tensor
.
blas
.
gemm_inplace
for
node
in
topo
if
isinstance
(
node
.
op
,
tensor
.
blas
.
Gemm
))
assert
all
(
node
.
op
==
tensor
.
blas
.
gemm_inplace
for
node
in
topo
if
isinstance
(
node
.
op
,
tensor
.
blas
.
Gemm
))
...
@@ -562,7 +562,7 @@ def makeSharedTester(shared_constructor_,
...
@@ -562,7 +562,7 @@ def makeSharedTester(shared_constructor_,
+
s_shared_specify
)])
+
s_shared_specify
)])
topo
=
f
.
maker
.
fgraph
.
toposort
()
topo
=
f
.
maker
.
fgraph
.
toposort
()
shp
=
f
()
shp
=
f
()
assert
n
umpy
.
all
(
shp
==
(
40
,
40
))
assert
n
p
.
all
(
shp
==
(
40
,
40
))
if
theano
.
config
.
mode
!=
'FAST_COMPILE'
:
if
theano
.
config
.
mode
!=
'FAST_COMPILE'
:
assert
sum
([
node
.
op
.
__class__
.
__name__
in
[
"Gemm"
,
"GpuGemm"
,
"StructuredDot"
]
for
node
in
topo
])
==
1
assert
sum
([
node
.
op
.
__class__
.
__name__
in
[
"Gemm"
,
"GpuGemm"
,
"StructuredDot"
]
for
node
in
topo
])
==
1
assert
all
(
node
.
op
==
tensor
.
blas
.
gemm_inplace
for
node
in
topo
if
isinstance
(
node
.
op
,
tensor
.
blas
.
Gemm
))
assert
all
(
node
.
op
==
tensor
.
blas
.
gemm_inplace
for
node
in
topo
if
isinstance
(
node
.
op
,
tensor
.
blas
.
Gemm
))
...
@@ -578,9 +578,9 @@ def makeSharedTester(shared_constructor_,
...
@@ -578,9 +578,9 @@ def makeSharedTester(shared_constructor_,
shp
=
(
1024
,
1024
)
shp
=
(
1024
,
1024
)
# Test the case with all zeros element
# Test the case with all zeros element
rng
=
n
umpy
.
random
.
RandomState
(
utt
.
fetch_seed
())
rng
=
n
p
.
random
.
RandomState
(
utt
.
fetch_seed
())
for
x
in
[
n
umpy
.
asarray
(
rng
.
rand
(
*
shp
),
dtype
=
dtype
),
for
x
in
[
n
p
.
asarray
(
rng
.
rand
(
*
shp
),
dtype
=
dtype
),
n
umpy
.
zeros
(
shp
,
dtype
=
dtype
)]:
n
p
.
zeros
(
shp
,
dtype
=
dtype
)]:
zeros
=
(
x
==
0
)
.
all
()
zeros
=
(
x
==
0
)
.
all
()
x
=
self
.
cast_value
(
x
)
x
=
self
.
cast_value
(
x
)
x_shared
=
self
.
shared_constructor
(
x
,
borrow
=
True
)
x_shared
=
self
.
shared_constructor
(
x
,
borrow
=
True
)
...
@@ -592,7 +592,7 @@ def makeSharedTester(shared_constructor_,
...
@@ -592,7 +592,7 @@ def makeSharedTester(shared_constructor_,
assert
x_shared
.
type
.
values_eq
(
x
,
x
)
assert
x_shared
.
type
.
values_eq
(
x
,
x
)
assert
x_shared
.
type
.
values_eq_approx
(
x
,
x
)
assert
x_shared
.
type
.
values_eq_approx
(
x
,
x
)
if
not
zeros
:
if
not
zeros
:
assert
not
n
umpy
.
allclose
(
self
.
ref_fct
(
x
),
self
.
ref_fct
(
y
))
assert
not
n
p
.
allclose
(
self
.
ref_fct
(
x
),
self
.
ref_fct
(
y
))
assert
not
x_shared
.
type
.
values_eq
(
x
,
y
)
assert
not
x_shared
.
type
.
values_eq
(
x
,
y
)
assert
not
x_shared
.
type
.
values_eq_approx
(
x
,
y
)
assert
not
x_shared
.
type
.
values_eq_approx
(
x
,
y
)
...
@@ -612,11 +612,11 @@ test_shared_options = makeSharedTester(
...
@@ -612,11 +612,11 @@ test_shared_options = makeSharedTester(
set_value_inplace_
=
False
,
set_value_inplace_
=
False
,
set_cast_value_inplace_
=
False
,
set_cast_value_inplace_
=
False
,
shared_constructor_accept_ndarray_
=
True
,
shared_constructor_accept_ndarray_
=
True
,
internal_type_
=
n
umpy
.
ndarray
,
internal_type_
=
n
p
.
ndarray
,
test_internal_type_
=
lambda
a
:
isinstance
(
a
,
n
umpy
.
ndarray
),
test_internal_type_
=
lambda
a
:
isinstance
(
a
,
n
p
.
ndarray
),
theano_fct_
=
lambda
a
:
a
*
2
,
theano_fct_
=
lambda
a
:
a
*
2
,
ref_fct_
=
lambda
a
:
n
umpy
.
asarray
((
a
*
2
)),
ref_fct_
=
lambda
a
:
n
p
.
asarray
((
a
*
2
)),
cast_value_
=
n
umpy
.
asarray
,
cast_value_
=
n
p
.
asarray
,
op_by_matrix_
=
False
,
op_by_matrix_
=
False
,
name
=
'test_shared_options'
)
name
=
'test_shared_options'
)
...
@@ -624,4 +624,4 @@ test_shared_options = makeSharedTester(
...
@@ -624,4 +624,4 @@ test_shared_options = makeSharedTester(
def
test_scalar_shared_options
():
def
test_scalar_shared_options
():
# Simple test to make sure we do not loose that fonctionality.
# Simple test to make sure we do not loose that fonctionality.
theano
.
shared
(
value
=
0.
,
name
=
'lk'
,
borrow
=
True
)
theano
.
shared
(
value
=
0.
,
name
=
'lk'
,
borrow
=
True
)
theano
.
shared
(
value
=
n
umpy
.
float32
(
0.
),
name
=
'lk'
,
borrow
=
True
)
theano
.
shared
(
value
=
n
p
.
float32
(
0.
),
name
=
'lk'
,
borrow
=
True
)
编写
预览
Markdown
格式
0%
重试
或
添加新文件
添加附件
取消
您添加了
0
人
到此讨论。请谨慎行事。
请先完成此评论的编辑!
取消
请
注册
或者
登录
后发表评论