Skip to content
项目
群组
代码片段
帮助
当前项目
正在载入...
登录 / 注册
切换导航面板
P
pytensor
项目
项目
详情
活动
周期分析
仓库
仓库
文件
提交
分支
标签
贡献者
图表
比较
统计图
议题
0
议题
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
CI / CD
CI / CD
流水线
作业
日程
统计图
Wiki
Wiki
代码片段
代码片段
成员
成员
折叠边栏
关闭边栏
活动
图像
聊天
创建新问题
作业
提交
问题看板
Open sidebar
testgroup
pytensor
Commits
e4c6880e
提交
e4c6880e
authored
7月 19, 2011
作者:
Razvan Pascanu
浏览文件
操作
浏览文件
下载
差异文件
merge; no conflicts
上级
594a136d
ce9a0ec5
显示空白字符变更
内嵌
并排
正在显示
8 个修改的文件
包含
1056 行增加
和
649 行删除
+1056
-649
basic_ops.py
theano/sandbox/cuda/basic_ops.py
+27
-0
__init__.py
theano/tensor/__init__.py
+1
-0
basic.py
theano/tensor/basic.py
+95
-597
elemwise.py
theano/tensor/elemwise.py
+86
-21
nnet.py
theano/tensor/nnet/nnet.py
+7
-0
opt.py
theano/tensor/opt.py
+5
-0
tensor_grad.py
theano/tensor/tensor_grad.py
+606
-0
test_rop.py
theano/tensor/tests/test_rop.py
+229
-31
没有找到文件。
theano/sandbox/cuda/basic_ops.py
浏览文件 @
e4c6880e
...
@@ -50,7 +50,20 @@ class HostFromGpu(Op):
...
@@ -50,7 +50,20 @@ class HostFromGpu(Op):
z
[
0
]
=
numpy
.
asarray
(
x
)
z
[
0
]
=
numpy
.
asarray
(
x
)
def
grad
(
self
,
inputs
,
grads
):
def
grad
(
self
,
inputs
,
grads
):
gz
,
=
grads
gz
,
=
grads
if
isinstance
(
gz
,
tensor
.
TensorType
):
# This would only happen if you call Lop, and provide a tensor
# that is not cuda
# This might require another look to be sure
return
[
gpu_from_host
(
gz
)]
return
[
gpu_from_host
(
gz
)]
else
:
return
[
gz
]
def
R_op
(
self
,
inputs
,
eval_points
):
ev
,
=
eval_points
if
isinstance
(
ev
,
tensor
.
TensorType
):
return
[
gpu_from_host
(
ev
)]
else
:
return
[
ev
]
def
infer_shape
(
self
,
node
,
xshp
):
def
infer_shape
(
self
,
node
,
xshp
):
return
xshp
return
xshp
host_from_gpu
=
HostFromGpu
()
host_from_gpu
=
HostFromGpu
()
...
@@ -72,7 +85,21 @@ class GpuFromHost(Op):
...
@@ -72,7 +85,21 @@ class GpuFromHost(Op):
z
[
0
]
=
type_support_filter
(
theano
.
_asarray
(
x
,
dtype
=
'float32'
),
tuple
([
0
]
*
x
.
ndim
),
0
,
z
[
0
])
z
[
0
]
=
type_support_filter
(
theano
.
_asarray
(
x
,
dtype
=
'float32'
),
tuple
([
0
]
*
x
.
ndim
),
0
,
z
[
0
])
def
grad
(
self
,
inputs
,
grads
):
def
grad
(
self
,
inputs
,
grads
):
gz
,
=
grads
gz
,
=
grads
if
isinstance
(
gz
,
CudaNdarrayType
):
# This would only happen if you call Lop, and provide a tensor
# that is not cuda
# This might require another look to be sure
return
[
host_from_gpu
(
gz
)]
return
[
host_from_gpu
(
gz
)]
else
:
return
[
gz
]
def
R_op
(
self
,
inputs
,
eval_points
):
ev
,
=
eval_points
if
isinstance
(
ev
,
CudaNdarrayType
):
return
[
host_from_gpu
(
ev
)]
else
:
return
[
ev
]
def
infer_shape
(
self
,
node
,
xshp
):
def
infer_shape
(
self
,
node
,
xshp
):
return
xshp
return
xshp
gpu_from_host
=
GpuFromHost
()
gpu_from_host
=
GpuFromHost
()
...
...
theano/tensor/__init__.py
浏览文件 @
e4c6880e
...
@@ -24,5 +24,6 @@ from sharedvar import tensor_constructor as shared
...
@@ -24,5 +24,6 @@ from sharedvar import tensor_constructor as shared
import
nnet
# used for softmax, sigmoid, etc.
import
nnet
# used for softmax, sigmoid, etc.
from
tensor_grad
import
Rop
,
Lop
,
grad
,
numeric_grad
,
verify_grad
theano/tensor/basic.py
浏览文件 @
e4c6880e
...
@@ -15,7 +15,7 @@ import numpy, theano
...
@@ -15,7 +15,7 @@ import numpy, theano
from
theano
import
gof
,
shared
from
theano
import
gof
,
shared
from
theano.gof
import
Apply
,
Constant
,
Op
,
Type
,
Value
,
Variable
from
theano.gof
import
Apply
,
Constant
,
Op
,
Type
,
Value
,
Variable
from
theano
import
gradient
import
elemwise
import
elemwise
from
theano
import
scalar
as
scal
from
theano
import
scalar
as
scal
...
@@ -1586,7 +1586,9 @@ class ScalarFromTensor(Op):
...
@@ -1586,7 +1586,9 @@ class ScalarFromTensor(Op):
return
[
tensor_from_scalar
(
dt
)]
return
[
tensor_from_scalar
(
dt
)]
def
R_op
(
self
,
inputs
,
eval_points
):
def
R_op
(
self
,
inputs
,
eval_points
):
if
None
in
eval_points
:
return
[
None
]
return
[
None
]
return
self
.
make_node
(
*
eval_points
)
.
outputs
def
__str__
(
self
):
def
__str__
(
self
):
return
self
.
__class__
.
__name__
return
self
.
__class__
.
__name__
...
@@ -1710,6 +1712,9 @@ class Shape(Op):
...
@@ -1710,6 +1712,9 @@ class Shape(Op):
out
[
0
]
=
theano
.
_asarray
(
x
.
shape
,
dtype
=
'int64'
)
out
[
0
]
=
theano
.
_asarray
(
x
.
shape
,
dtype
=
'int64'
)
def
grad
(
self
,
inp
,
grads
):
def
grad
(
self
,
inp
,
grads
):
return
[
None
]
return
[
None
]
def
R_op
(
self
,
inputs
,
eval_points
):
return
[
None
]
@constructor
@constructor
def
old_shape
(
a
):
def
old_shape
(
a
):
"""Return the shape tuple of a TensorType Variable, it may be either symbolic or nonsymbolic.
"""Return the shape tuple of a TensorType Variable, it may be either symbolic or nonsymbolic.
...
@@ -1788,6 +1793,10 @@ class SpecifyShape(Op):
...
@@ -1788,6 +1793,10 @@ class SpecifyShape(Op):
return
[
specify_shape
(
gz
,
s
),
None
]
return
[
specify_shape
(
gz
,
s
),
None
]
def
R_op
(
self
,
inputs
,
eval_points
):
def
R_op
(
self
,
inputs
,
eval_points
):
if
eval_points
[
0
]
is
None
:
# It means that the this op sits on top of a non-differentiable
# path
return
[
None
]
return
self
.
make_node
(
eval_points
[
0
],
*
inputs
[
1
:])
.
outputs
return
self
.
make_node
(
eval_points
[
0
],
*
inputs
[
1
:])
.
outputs
specify_shape
=
SpecifyShape
()
specify_shape
=
SpecifyShape
()
...
@@ -1868,6 +1877,12 @@ class MaxAndArgmax(Op):
...
@@ -1868,6 +1877,12 @@ class MaxAndArgmax(Op):
rval
=
tuple
([
ishape
[
i
]
for
(
i
,
b
)
in
enumerate
(
node
.
inputs
[
0
]
.
type
.
broadcastable
)
if
i
!=
axis
.
data
])
rval
=
tuple
([
ishape
[
i
]
for
(
i
,
b
)
in
enumerate
(
node
.
inputs
[
0
]
.
type
.
broadcastable
)
if
i
!=
axis
.
data
])
return
[
rval
,
rval
]
return
[
rval
,
rval
]
def
R_op
(
self
,
inputs
,
eval_points
):
if
eval_points
[
0
]
is
None
:
return
[
None
,
None
]
return
[
self
.
make_node
(
eval_points
[
0
],
inputs
[
1
])
.
outputs
[
0
],
None
]
def
grad
(
self
,
inp
,
grads
):
def
grad
(
self
,
inp
,
grads
):
# @warning: This only works if axis is 0, else the max is
# @warning: This only works if axis is 0, else the max is
# broadcasted wrong in the call to eq.
# broadcasted wrong in the call to eq.
...
@@ -2474,6 +2489,8 @@ class Alloc(gof.Op):
...
@@ -2474,6 +2489,8 @@ class Alloc(gof.Op):
pass
pass
return
ret
return
ret
def
R_op
(
self
,
inputs
,
eval_points
):
return
[
None
]
alloc
=
Alloc
()
alloc
=
Alloc
()
pprint
.
assign
(
alloc
,
printing
.
FunctionPrinter
(
'alloc'
))
pprint
.
assign
(
alloc
,
printing
.
FunctionPrinter
(
'alloc'
))
...
@@ -3341,6 +3358,13 @@ class Subtensor(Op):
...
@@ -3341,6 +3358,13 @@ class Subtensor(Op):
hv
=
self
.
helper_c_code_cache_version
()
hv
=
self
.
helper_c_code_cache_version
()
return
(
1
,
hv
)
return
(
1
,
hv
)
def
R_op
(
self
,
inputs
,
eval_points
):
# Subtensor is not differentiable wrt to its indices, therefore we
# do not even need to consider the eval_points provided for those
# (they should be defaulted to zeros_like by the global R_op)
if
eval_points
[
0
]
is
None
:
return
[
None
]
return
self
.
make_node
(
eval_points
[
0
],
*
inputs
[
1
:])
.
outputs
class
SubtensorPrinter
:
class
SubtensorPrinter
:
...
@@ -3646,6 +3670,14 @@ class IncSubtensor(Op):
...
@@ -3646,6 +3670,14 @@ class IncSubtensor(Op):
def
infer_shape
(
self
,
node
,
shapes
):
def
infer_shape
(
self
,
node
,
shapes
):
return
[
shapes
[
0
]]
return
[
shapes
[
0
]]
def
R_op
(
self
,
inputs
,
eval_points
):
if
eval_points
[
0
]
is
None
or
eval_points
[
1
]
is
None
:
return
[
None
]
# Again we ignore eval points for indices because incsubtensor is
# not differentiable wrt to those
return
self
.
make_node
(
eval_points
[
0
],
eval_points
[
1
],
*
inputs
[
2
:])
.
outputs
def
grad
(
self
,
inputs
,
grads
):
def
grad
(
self
,
inputs
,
grads
):
g_output
,
=
grads
g_output
,
=
grads
x
,
y
=
inputs
[:
2
]
x
,
y
=
inputs
[:
2
]
...
@@ -3756,6 +3788,10 @@ class Split(Op):
...
@@ -3756,6 +3788,10 @@ class Split(Op):
_
,
axis
,
_
=
inputs
_
,
axis
,
_
=
inputs
return
[
join
(
axis
,
*
g_outputs
),
None
,
None
]
return
[
join
(
axis
,
*
g_outputs
),
None
,
None
]
def
R_op
(
self
,
inputs
,
eval_points
):
if
eval_points
[
0
]
is
None
:
return
[
None
for
i
in
self
.
len_splits
]
return
self
.
make_node
(
eval_points
[
0
],
*
inputs
[
1
:])
.
outputs
class
Rebroadcast
(
Op
):
class
Rebroadcast
(
Op
):
"""
"""
...
@@ -3816,6 +3852,13 @@ class Rebroadcast(Op):
...
@@ -3816,6 +3852,13 @@ class Rebroadcast(Op):
return
[
tuple
(
l
)]
return
[
tuple
(
l
)]
def
R_op
(
self
,
inputs
,
eval_points
):
if
eval_points
[
0
]
is
None
:
return
[
None
]
return
self
.
make_node
(
*
eval_points
)
.
outputs
def
addbroadcast
(
x
,
*
axes
):
def
addbroadcast
(
x
,
*
axes
):
"""
"""
Make the input broadcastable in the specified axes.
Make the input broadcastable in the specified axes.
...
@@ -3968,6 +4011,11 @@ class Join(Op):
...
@@ -3968,6 +4011,11 @@ class Join(Op):
out
[
0
]
=
theano
.
_asarray
(
numpy
.
concatenate
(
tensors
,
axis
=
axis
),
out
[
0
]
=
theano
.
_asarray
(
numpy
.
concatenate
(
tensors
,
axis
=
axis
),
dtype
=
node
.
outputs
[
0
]
.
type
.
dtype
)
dtype
=
node
.
outputs
[
0
]
.
type
.
dtype
)
def
R_op
(
self
,
inputs
,
eval_points
):
if
None
in
eval_points
[
1
:]:
return
[
None
]
return
self
.
make_node
(
inputs
[
0
],
*
eval_points
[
1
:])
.
outputs
def
grad
(
self
,
axis_and_tensors
,
grads
):
def
grad
(
self
,
axis_and_tensors
,
grads
):
""" The gradient wrt a join op is a `Split`, used to partition the gradient along the
""" The gradient wrt a join op is a `Split`, used to partition the gradient along the
`axis` which was used for joining.
`axis` which was used for joining.
...
@@ -4299,6 +4347,13 @@ class Reshape(Op):
...
@@ -4299,6 +4347,13 @@ class Reshape(Op):
x
,
shp
=
inp
x
,
shp
=
inp
g_out
,
=
grads
g_out
,
=
grads
return
[
reshape
(
g_out
,
shape
(
x
),
ndim
=
x
.
ndim
),
None
]
return
[
reshape
(
g_out
,
shape
(
x
),
ndim
=
x
.
ndim
),
None
]
def
R_op
(
self
,
inputs
,
eval_points
):
if
eval_points
[
0
]
is
None
:
return
[
None
]
return
self
.
make_node
(
eval_points
[
0
],
*
inputs
[
1
:])
.
outputs
def
infer_shape
(
self
,
node
,
ishapes
):
def
infer_shape
(
self
,
node
,
ishapes
):
# inputs[1] can contain at most one value of '-1', meaning the actual
# inputs[1] can contain at most one value of '-1', meaning the actual
# shape of the output will be automatically computed by reshape, so
# shape of the output will be automatically computed by reshape, so
...
@@ -4376,6 +4431,11 @@ class Flatten(Op):
...
@@ -4376,6 +4431,11 @@ class Flatten(Op):
g_out
,
=
grads
g_out
,
=
grads
return
[
reshape
(
g_out
,
shape
(
x
),
x
.
ndim
)]
return
[
reshape
(
g_out
,
shape
(
x
),
x
.
ndim
)]
def
R_op
(
self
,
inputs
,
eval_points
):
if
None
in
eval_points
:
return
[
None
]
return
self
.
make_node
(
*
eval_points
)
.
outputs
def
flatten
(
x
,
outdim
=
1
):
def
flatten
(
x
,
outdim
=
1
):
return
Flatten
(
outdim
)(
x
)
return
Flatten
(
outdim
)(
x
)
...
@@ -4501,6 +4561,8 @@ class ARange(Op):
...
@@ -4501,6 +4561,8 @@ class ARange(Op):
gz
,
=
grads
gz
,
=
grads
return
[
None
]
*
len
(
inputs
)
return
[
None
]
*
len
(
inputs
)
def
R_op
(
self
,
inputs
,
eval_points
):
return
[
None
]
_arange
=
{}
_arange
=
{}
def
arange
(
start
,
stop
=
None
,
step
=
1
,
dtype
=
None
):
def
arange
(
start
,
stop
=
None
,
step
=
1
,
dtype
=
None
):
# If only one argument is provided, it is in fact the "stop" argument,
# If only one argument is provided, it is in fact the "stop" argument,
...
@@ -4748,6 +4810,11 @@ class AdvancedSubtensor1(Op):
...
@@ -4748,6 +4810,11 @@ class AdvancedSubtensor1(Op):
assert
len
(
inputs
)
==
2
assert
len
(
inputs
)
==
2
return
[
advanced_inc_subtensor1
(
zeros_like
(
inputs
[
0
]),
gz
,
inputs
[
1
])]
+
[
None
]
*
(
len
(
inputs
)
-
1
)
return
[
advanced_inc_subtensor1
(
zeros_like
(
inputs
[
0
]),
gz
,
inputs
[
1
])]
+
[
None
]
*
(
len
(
inputs
)
-
1
)
def
R_op
(
self
,
inputs
,
eval_points
):
if
eval_points
[
0
]
is
None
:
return
[
None
]
return
self
.
make_node
(
eval_points
[
0
],
*
inputs
[
1
:])
.
outputs
def
infer_shape
(
self
,
node
,
ishapes
):
def
infer_shape
(
self
,
node
,
ishapes
):
x
,
ilist
=
ishapes
x
,
ilist
=
ishapes
return
[
ilist
+
x
[
1
:]]
return
[
ilist
+
x
[
1
:]]
...
@@ -4802,6 +4869,13 @@ class AdvancedIncSubtensor1(Op):
...
@@ -4802,6 +4869,13 @@ class AdvancedIncSubtensor1(Op):
x
,
y
,
ilist
=
ishapes
x
,
y
,
ilist
=
ishapes
return
[
x
]
return
[
x
]
def
R_op
(
self
,
inputs
,
eval_points
):
if
None
in
eval_points
[:
2
]:
return
[
None
]
return
self
.
make_node
(
eval_points
[
0
],
eval_points
[
1
],
*
inputs
[
2
:])
.
outputs
def
grad
(
self
,
inputs
,
grads
):
def
grad
(
self
,
inputs
,
grads
):
g_output
,
=
grads
g_output
,
=
grads
x
,
y
=
inputs
[:
2
]
x
,
y
=
inputs
[:
2
]
...
@@ -4854,6 +4928,11 @@ class AdvancedSubtensor(Op):
...
@@ -4854,6 +4928,11 @@ class AdvancedSubtensor(Op):
raise
NotImplementedError
(
'Advanced indexing of x with arguments (
%
s) not supported yet'
\
raise
NotImplementedError
(
'Advanced indexing of x with arguments (
%
s) not supported yet'
\
%
','
.
join
(
str
(
input
)
for
input
in
inputs
))
%
','
.
join
(
str
(
input
)
for
input
in
inputs
))
def
R_op
(
self
,
inputs
,
eval_points
):
if
eval_points
[
0
]
is
None
:
return
[
None
]
return
self
.
make_node
(
eval_points
[
0
],
*
inputs
[
1
:])
.
outputs
def
infer_shape
(
self
,
node
,
ishapes
):
def
infer_shape
(
self
,
node
,
ishapes
):
# Really special case
# Really special case
if
len
(
ishapes
)
==
3
:
if
len
(
ishapes
)
==
3
:
...
@@ -4917,6 +4996,11 @@ class AdvancedIncSubtensor(Op):
...
@@ -4917,6 +4996,11 @@ class AdvancedIncSubtensor(Op):
# grad on x is grad on output
# grad on x is grad on output
# grad on y is grad_output[idx_list]
# grad on y is grad_output[idx_list]
# grad on rest is None
# grad on rest is None
def
R_op
(
self
,
inputs
,
eval_points
):
if
None
in
eval_points
[:
2
]:
return
[
None
]
return
self
.
make_node
(
eval_points
[
0
],
eval_points
[
1
],
*
inputs
[
2
:])
.
outputs
...
@@ -5018,6 +5102,16 @@ class Dot(Op):
...
@@ -5018,6 +5102,16 @@ class Dot(Op):
rval
=
dot
(
gz
,
y
.
T
),
dot
(
x
.
T
,
gz
)
rval
=
dot
(
gz
,
y
.
T
),
dot
(
x
.
T
,
gz
)
return
cast
(
rval
[
0
],
x
.
dtype
),
cast
(
rval
[
1
],
y
.
dtype
)
return
cast
(
rval
[
0
],
x
.
dtype
),
cast
(
rval
[
1
],
y
.
dtype
)
def
R_op
(
self
,
inputs
,
eval_points
):
# R_op for a \dot b evaluted at c for a and d for b is
# simply c \dot b + a \dot d
if
None
in
eval_points
:
return
[
None
]
t1
=
self
.
make_node
(
eval_points
[
0
],
inputs
[
1
])
.
outputs
[
0
]
t2
=
self
.
make_node
(
inputs
[
0
],
eval_points
[
1
])
.
outputs
[
0
]
return
[
t1
+
t2
]
def
infer_shape
(
self
,
node
,
shapes
):
def
infer_shape
(
self
,
node
,
shapes
):
xshp
,
yshp
=
shapes
xshp
,
yshp
=
shapes
x
,
y
=
node
.
inputs
x
,
y
=
node
.
inputs
...
@@ -5235,600 +5329,4 @@ class Outer(Op):
...
@@ -5235,600 +5329,4 @@ class Outer(Op):
return
"outer"
return
"outer"
outer
=
Outer
()
outer
=
Outer
()
########################
# R Operator
########################
def
Rop
(
f
,
wrt
,
eval_points
):
"""
Computes the R operation on `f` wrt to `wrt` evaluated at points given
in `eval_points`. Mathematically this stands for the jacobian of `f` wrt
to `wrt` right muliplied by the eval points.
:type f: `Variable` or list of `Variable`s
`f` stands for the output of the computational graph to which you
want to apply the R operator
:type wrt: `Variable` or list of `Variables`s
variables for which you compute the R operator of the expression
described by `f`
:type eval_points: `Variable` or list of `Variable`s
evalutation points for each of the variables in `wrt`
:rtype: `Variable` or list of `Variable`s depending on type of f
:return: symbolic expression such that
R_op[i] = sum_j ( d f[i] / d wrt[j]) eval_point[j]
where the indices in that expression are magic multidimensional
indices that specify both the position within a list and all
coordinates of the tensor element in the last
"""
if
not
isinstance
(
wrt
,
(
list
,
tuple
)):
wrt
=
[
wrt
]
if
not
isinstance
(
eval_points
,
(
list
,
tuple
)):
eval_points
=
[
eval_points
]
if
not
isinstance
(
f
,
(
list
,
tuple
)):
f
=
[
f
]
assert
len
(
wrt
)
==
len
(
eval_points
)
seen_nodes
=
{}
def
_traverse
(
node
):
if
node
is
None
:
return
None
else
:
op
=
node
.
op
inputs
=
node
.
inputs
if
not
hasattr
(
op
,
'R_op'
):
raise
Exception
((
' R_op was not implemented for
%
s'
' operation. Email the mailing list'
' for help'
)
%
op
.
__class__
.
__name__
)
# Compute the evaluation points corresponding to each of the
# inputs of the node
local_eval_points
=
[]
for
inp
in
inputs
:
if
inp
in
wrt
:
local_eval_points
.
append
(
eval_points
[
wrt
.
index
(
inp
)]
)
elif
inp
.
owner
is
None
:
local_eval_points
.
append
(
zeros_like
(
inp
)
)
elif
inp
.
owner
in
seen_nodes
:
local_eval_points
.
append
(
seen_nodes
[
inp
.
owner
][
inp
.
owner
.
outputs
.
index
(
inp
)
]
)
else
:
# We actually need to compute the R_op for this node
_traverse
(
inp
.
owner
)
local_eval_points
.
append
(
seen_nodes
[
inp
.
owner
][
inp
.
owner
.
outputs
.
index
(
inp
)
])
from
theano.sandbox
import
cuda
if
cuda
.
cuda_available
:
from
theano.sandbox.cuda.basic_ops
import
gpu_from_host
,
host_from_gpu
from
theano.sandbox.cuda.type
import
CudaNdarrayType
for
idx
,
(
x
,
y
)
in
enumerate
(
zip
(
inputs
,
local_eval_points
)):
if
x
.
type
!=
y
.
type
:
if
(
isinstance
(
x
.
type
,
CudaNdarrayType
)
and
isinstance
(
y
.
type
,
TensorType
)):
assert
x
.
type
.
ndim
==
y
.
type
.
ndim
assert
y
.
type
.
dtype
==
'float32'
elif
(
isinstance
(
x
.
type
,
TensorType
)
and
isinstance
(
y
.
type
,
CudaNdarrayType
)):
assert
x
.
type
.
ndim
==
y
.
type
.
ndim
assert
x
.
type
.
dtype
==
'float32'
else
:
assert
x
.
type
==
y
.
type
else
:
for
x
,
y
in
zip
(
inputs
,
local_eval_points
):
assert
x
.
type
==
y
.
type
seen_nodes
[
node
]
=
op
.
R_op
(
node
.
inputs
,
local_eval_points
)
return
None
# Populate the dictionary
for
out
in
f
:
_traverse
(
out
.
owner
)
rval
=
[]
for
out
in
f
:
if
out
in
wrt
:
rval
.
append
(
eval_points
[
wrt
.
index
(
out
)])
elif
seen_nodes
[
out
.
owner
][
out
.
owner
.
outputs
.
index
(
out
)]
is
None
:
raise
ValueError
((
'The function is not differentiable with '
'respect to the provided inputs !'
))
else
:
rval
.
append
(
seen_nodes
[
out
.
owner
][
out
.
owner
.
outputs
.
index
(
out
)]
)
if
len
(
rval
)
==
1
:
return
rval
[
0
]
else
:
return
rval
def
Lop
(
f
,
wrt
,
eval_points
,
consider_constant
=
[],
warn_type
=
False
,
disconnected_inputs
=
'raise'
):
"""
Computes the L operation on `f` wrt to `wrt` evaluated at points given
in `eval_points`. Mathematically this stands for the jacobian of `f` wrt
to `wrt` left muliplied by the eval points.
:type f: `Variable` or list of `Variable`s
`f` stands for the output of the computational graph to which you
want to apply the L operator
:type wrt: `Variable` or list of `Variables`s
variables for which you compute the L operator of the expression
described by `f`
:type eval_points: `Variable` or list of `Variable`s
evalutation points for each of the variables in `f`
:rtype: `Variable` or list of `Variable`s depending on type of f
:return: symbolic expression such that
L_op[i] = sum_i ( d f[i] / d wrt[j]) eval_point[i]
where the indices in that expression are magic multidimensional
indices that specify both the position within a list and all
coordinates of the tensor element in the last
"""
if
not
isinstance
(
f
,
TensorVariable
):
raise
TypeError
(
'In tensor.Lop(), cost argument should be a TensorVariable.'
,
f
)
if
type
(
eval_points
)
not
in
(
list
,
tuple
):
eval_points
=
[
eval_points
]
if
type
(
f
)
not
in
(
list
,
tuple
):
f
=
[
f
]
inputs
=
gof
.
graph
.
inputs
(
f
)
gmap
=
gradient
.
grad_sources_inputs
(
zip
(
f
,
eval_points
),
list
(
inputs
)
+
list
(
consider_constant
),
warn_type
=
warn_type
)
# Note : If p is not in gmap there can be several reasons, among which
# is the fact that p might not be part of the computational graph. A
# simple example is that for a+b for e.g. a[0] is not part of the graph,
# so Theano does not know how to compute TT.grad(TT.sum(a+b), a[0])
# such subtle cases can be fixed by a more careful implementation of the
# gradient, but for now Theano needs to throw an exception, and make the
# user aware that it does not know how to compute that gradient
if
not
isinstance
(
wrt
,
(
list
,
tuple
)):
wrt
=
[
wrt
]
ret
=
[]
for
p
in
wrt
:
if
p
in
gmap
:
ret
.
append
(
gmap
[
p
])
else
:
message
=
(
"Lop method was asked to compute the gradient "
"with respect to a variable that is not part of "
"the computational graph of the cost, or is used "
"only by a non-differentiable operator:
%
s"
%
p
)
if
disconnected_inputs
==
'ignore'
:
pass
elif
disconnected_inputs
==
'warn'
:
warnings
.
warn
(
message
,
stacklevel
=
1
)
elif
disconnected_inputs
==
'raise'
:
raise
ValueError
(
message
)
else
:
raise
ValueError
(
"Invalid value for keyword "
"'disconnected_inputs', valid values are "
"'ignore', 'warn' and 'raise'."
)
ret
.
append
(
zeros_like
(
p
))
if
len
(
ret
)
==
1
:
return
ret
[
0
]
else
:
return
ret
#########################
# Gradient
#########################
def
grad
(
cost
,
wrt
,
g_cost
=
None
,
consider_constant
=
[],
warn_type
=
False
,
disconnected_inputs
=
'raise'
):
"""
:type cost: Scalar (0-dimensional) `Variable`
:type wrt: `Variable` or list of `Variable`s.
:type g_cost: Scalar `Variable`, or None
:param g_cost: an expression for the gradient through cost. The default is
``ones_like(cost)``.
:param consider_constant: a list of expressions not to backpropagate through
:param warn_type: a value of True will cause warnings to be logged for any Op that emits a
gradient that does not match its input type.
:type disconnected_inputs: string
:param disconnected_inputs: Defines the behaviour if some of the variables
in ``wrt`` are not part of the computational graph computing ``cost``
(or if all links are non-differentiable). The possible values are:
- 'ignore': considers that the gradient on these parameters is zero.
- 'warn': consider the gradient zero, and print a warning.
- 'raise': raise an exception.
:rtype: `Variable` or list of `Variable`s (depending upon `wrt`)
:return: symbolic expression of gradient of `cost` with respect to `wrt`.
If `wrt` is a list, then return a list containing the gradient of `cost` wrt
each element of the list. If an element of `wrt` is not differentiable
with respect to the output, then a zero variable is returned.
This function is a wrapper around the more general function
`theano.gradient.grad_sources_inputs``.
"""
if
not
isinstance
(
cost
,
TensorVariable
):
raise
TypeError
(
'In tensor.grad(), cost argument should be a TensorVariable.'
,
cost
)
if
cost
.
type
.
ndim
:
raise
TypeError
(
'In tensor.grad, "cost" argument should be a scalar, but ndim'
' is
%
i (should be 0). If you want to compute the gradient of'
' the sum of cost, you should use cost.sum().'
%
cost
.
type
.
ndim
)
if
g_cost
is
None
:
g_cost
=
ones_like
(
cost
)
inputs
=
gof
.
graph
.
inputs
([
cost
])
gmap
=
gradient
.
grad_sources_inputs
(
[(
cost
,
g_cost
)],
list
(
inputs
)
+
list
(
consider_constant
),
warn_type
=
warn_type
)
# Note : If p is not in gmap there can be several reasons, among which
# is the fact that p might not be part of the computational graph. A
# simple example is that for a+b for e.g. a[0] is not part of the graph,
# so Theano does not know how to compute TT.grad(TT.sum(a+b), a[0])
# such subtle cases can be fixed by a more careful implementation of the
# gradient, but for now Theano needs to throw an exception, and make the
# user aware that it does not know how to compute that gradient
if
not
isinstance
(
wrt
,
(
list
,
tuple
)):
wrt
=
[
wrt
]
ret
=
[]
for
p
in
wrt
:
if
p
in
gmap
:
ret
.
append
(
gmap
[
p
])
else
:
message
=
(
"grad method was asked to compute the gradient "
"with respect to a variable that is not part of "
"the computational graph of the cost, or is used "
"only by a non-differentiable operator:
%
s"
%
p
)
if
disconnected_inputs
==
'ignore'
:
pass
elif
disconnected_inputs
==
'warn'
:
warnings
.
warn
(
message
,
stacklevel
=
1
)
elif
disconnected_inputs
==
'raise'
:
raise
ValueError
(
message
)
else
:
raise
ValueError
(
"Invalid value for keyword "
"'disconnected_inputs', valid values are "
"'ignore', 'warn' and 'raise'."
)
ret
.
append
(
zeros_like
(
p
))
if
len
(
ret
)
==
1
:
return
ret
[
0
]
else
:
return
ret
class
numeric_grad
:
"""WRITEME"""
# Note on step sizes and tolerances:
#
# There is a relationship between the step size and the function value and the measurement
# error that is incurred due to rounding. The finite difference we measure is
# delta = f(x0) - f(x0+eps)
#
# For maximum precision, f should be close to zero.
# For every power of 2 that f departs from zero, we lose a bit of precision in delta.
#
# Even in this case of maximum accuracy, there is a tradeoff between stepsize and
# measurement error.
# Taking small steps allows us to measure large derivatives accuractly, but longer steps
# are required to measure small derivatives accurately. However longer steps introduce
# bias into our measurement in general for non-linear functions.
#
# It would be interesting to have a version of numeric grad that used an adaptive stepsize.
#
# For now, we use a heuristic that catches very bad gradients, but is not perfectly
# accurate.
type_eps
=
{
'float64'
:
1e-7
,
'float32'
:
3e-4
,
numpy
.
dtype
(
'float64'
):
1e-7
,
numpy
.
dtype
(
'float32'
):
3e-4
}
def
__init__
(
self
,
f
,
pt
,
eps
=
None
):
"""Return the gradient of f at pt.
This function computes the gradient by a one-sided finite differences of a
fixed step size (eps).
It is assumed that f(...) will return a scalar.
It is assumed that all f's inputs are numpy.ndarray objects.
:param eps: the stepsize for the finite differencing. None means input
dtype-dependent. See `type_eps`.
"""
def
prod
(
inputs
):
rval
=
1
for
i
in
inputs
:
rval
*=
i
return
rval
packed_pt
=
False
if
not
isinstance
(
pt
,
(
list
,
tuple
)):
pt
=
[
pt
]
packed_pt
=
True
apt
=
[
numpy
.
array
(
p
)
for
p
in
pt
]
shapes
=
[
p
.
shape
for
p
in
apt
]
dtypes
=
[
str
(
p
.
dtype
)
for
p
in
apt
]
# TODO: remove this eventually (why was this here in the first place ?)
# In the case of CSM, the arguments are a mixture of floats and integers...
#if not dtypes == [dtypes[0]] * len(apt):
#raise TypeError('All function arguments must have same dtype')
total_size
=
__builtin__
.
sum
(
prod
(
sh
)
for
sh
in
shapes
)
working_dtype
=
__builtin__
.
min
((
self
.
type_eps
[
dt
],
dt
)
for
dt
in
dtypes
)[
1
]
#create un-initialized memory
x
=
numpy
.
ndarray
((
total_size
,),
dtype
=
working_dtype
)
gx
=
numpy
.
ndarray
((
total_size
,),
dtype
=
working_dtype
)
if
eps
is
None
:
eps
=
__builtin__
.
max
(
self
.
type_eps
[
dt
]
for
dt
in
dtypes
)
#set up aliases so that apt[i] is backed by memory in x
# and self.gf is backed by memory in gx
cur_pos
=
0
self
.
gf
=
[]
for
i
,
p
in
enumerate
(
apt
):
p_size
=
prod
(
p
.
shape
)
# set up alias
apt
[
i
]
=
x
[
cur_pos
:
cur_pos
+
p_size
]
.
reshape
(
p
.
shape
)
self
.
gf
.
append
(
gx
[
cur_pos
:
cur_pos
+
p_size
]
.
reshape
(
p
.
shape
))
# initialize with p's value
apt
[
i
][
...
]
=
p
cur_pos
+=
p_size
f_x
=
f
(
*
[
p
.
copy
()
for
p
in
apt
])
# now iterate over the elements of x, and call f on apt.
x_copy
=
x
.
copy
()
for
i
in
xrange
(
total_size
):
x
[:]
=
x_copy
x
[
i
]
+=
eps
f_eps
=
f
(
*
apt
)
gx
[
i
]
=
numpy
.
asarray
((
f_eps
-
f_x
)
/
eps
)
if
packed_pt
:
self
.
gf
=
self
.
gf
[
0
]
@staticmethod
def
abs_rel_err
(
a
,
b
):
"""Return absolute and relative error between a and b.
The relative error is a small number when a and b are close, relative to how big they are.
Formulas used:
abs_err = abs(a - b)
rel_err = abs_err / (abs(a) + abs(b))
The tuple (abs_err, rel_err) is returned
"""
abs_err
=
abs
(
a
-
b
)
rel_err
=
abs_err
/
(
abs
(
a
)
+
abs
(
b
))
return
(
abs_err
,
rel_err
)
def
abs_rel_errors
(
self
,
g_pt
):
"""Return the abs and rel error of gradient estimate `g_pt`
`g_pt` must be a list of ndarrays of the same length as self.gf, otherwise a ValueError
is raised.
Corresponding ndarrays in `g_pt` and `self.gf` must have the same shape or ValueError
is raised.
"""
if
len
(
g_pt
)
!=
len
(
self
.
gf
):
raise
ValueError
(
'argument has wrong number of elements'
,
len
(
g_pt
))
errs
=
[]
for
i
,
(
a
,
b
)
in
enumerate
(
zip
(
g_pt
,
self
.
gf
)):
if
a
.
shape
!=
b
.
shape
:
raise
ValueError
(
'argument element
%
i has wrong shape
%
s'
%
(
i
,
str
((
a
.
shape
,
b
.
shape
))))
errs
.
append
(
numeric_grad
.
abs_rel_err
(
a
,
b
))
return
errs
def
max_err
(
self
,
g_pt
,
abs_tol
,
rel_tol
):
"""Find the biggest error between g_pt and self.gf.
What is measured is the violation of relative and absolute errors,
wrt the provided tolerances (abs_tol, rel_tol).
A value > 1 means both tolerances are exceeded.
Return the argmax of min(abs_err / abs_tol, rel_err / rel_tol) over g_pt,
as well as abs_err and rel_err at this point.
"""
pos
=
[]
errs
=
[]
abs_errs
=
[]
rel_errs
=
[]
abs_rel_errs
=
self
.
abs_rel_errors
(
g_pt
)
for
abs_err
,
rel_err
in
abs_rel_errs
:
scaled_err
=
numpy
.
minimum
(
abs_err
/
abs_tol
,
rel_err
/
rel_tol
)
max_i
=
scaled_err
.
argmax
()
pos
.
append
(
max_i
)
errs
.
append
(
scaled_err
.
flatten
()[
max_i
])
abs_errs
.
append
(
abs_err
.
flatten
()[
max_i
])
rel_errs
.
append
(
rel_err
.
flatten
()[
max_i
])
# max over the arrays in g_pt
max_arg
=
numpy
.
argmax
(
errs
)
max_pos
=
pos
[
max_arg
]
return
(
max_arg
,
pos
[
max_arg
],
abs_errs
[
max_arg
],
rel_errs
[
max_arg
])
def
verify_grad
(
fun
,
pt
,
n_tests
=
2
,
rng
=
None
,
eps
=
None
,
abs_tol
=
None
,
rel_tol
=
None
,
mode
=
None
,
cast_to_output_type
=
False
):
""" Test a gradient by Finite Difference Method. Raise error on failure.
Example:
>>> verify_grad(theano.tensor.tanh,
(numpy.asarray([[2,3,4], [-1, 3.3, 9.9]]),),
rng=numpy.random)
Raises an Exception if the difference between the analytic gradient and
numerical gradient (computed through the Finite Difference Method) of a random
projection of the fun's output to a scalar exceeds
the given tolerance.
:param fun: a Python function that takes Theano variables as inputs,
and returns a Theano variable. For instance, an Op instance with
a single output.
:param pt: the list of numpy.ndarrays to use as input values.
These arrays must be either float32 or float64 arrays.
:param n_tests: number of times to run the test
:param rng: random number generator used to sample u, we test gradient of sum(u * fun) at pt
:param eps: stepsize used in the Finite Difference Method (Default None is type-dependent)
:param abs_tol: absolute tolerance used as threshold for gradient comparison
:param rel_tol: relative tolerance used as threshold for gradient comparison
:note: WARNING to unit-test writers: if `op` is a function that builds a graph,
try to make it a SMALL graph. Often verify grad is run in
debug mode, which can be very slow if it has to verify a lot
of intermediate computations.
:note: This op does not support multiple outputs. In tests/test_scan.py there is
an experimental verify_grad that covers that case as well by using random
projections.
"""
assert
isinstance
(
pt
,
(
list
,
tuple
))
pt
=
[
numpy
.
array
(
p
)
for
p
in
pt
]
for
i
,
p
in
enumerate
(
pt
):
if
p
.
dtype
not
in
(
'float32'
,
'float64'
):
raise
TypeError
((
'verify_grad can work only with floating point '
'inputs, but input
%
i has dtype "
%
s".'
)
%
(
i
,
p
.
dtype
))
_type_tol
=
dict
(
# relativ error tolerances for different types
float32
=
1e-2
,
float64
=
1e-4
)
if
abs_tol
is
None
:
abs_tol
=
__builtin__
.
max
(
_type_tol
[
str
(
p
.
dtype
)]
for
p
in
pt
)
if
rel_tol
is
None
:
rel_tol
=
__builtin__
.
max
(
_type_tol
[
str
(
p
.
dtype
)]
for
p
in
pt
)
if
rng
is
None
:
raise
TypeError
(
'rng should be a valid instance of numpy.random.RandomState.'
,
'You may want to use theano.tests.unittest_tools.verify_grad instead of theano.tensor.verify_grad.'
)
# We allow input downcast in function, because numeric_grad works in the
# most precise dtype used among the inputs, so we may need to cast some.
def
function
(
inputs
,
output
):
if
mode
is
None
:
f
=
compile
.
function
(
inputs
,
output
,
accept_inplace
=
True
,
allow_input_downcast
=
True
)
else
:
f
=
compile
.
function
(
inputs
,
output
,
accept_inplace
=
True
,
allow_input_downcast
=
True
,
mode
=
mode
)
return
f
tensor_pt
=
[
TensorType
(
as_tensor_variable
(
p
)
.
dtype
,
as_tensor_variable
(
p
)
.
broadcastable
)(
name
=
'input
%
i'
%
i
)
for
i
,
p
in
enumerate
(
pt
)]
#fun can be either a function or an actual Op instance
o_output
=
fun
(
*
tensor_pt
)
if
isinstance
(
o_output
,
list
):
raise
NotImplementedError
(
'cant (yet) autotest gradient of fun with multiple outputs'
)
# we could make loop over outputs making random projections R for each,
# but this doesn't handle the case where not all the outputs are
# differentiable... so I leave this as TODO for now -JB.
o_fn
=
function
(
tensor_pt
,
o_output
)
o_fn_out
=
o_fn
(
*
[
p
.
copy
()
for
p
in
pt
])
if
isinstance
(
o_fn_out
,
tuple
)
or
isinstance
(
o_fn_out
,
list
):
raise
TypeError
(
'It seems like you are trying to use verify_grad '
'on an op or a function which outputs a list: there should'
' be a single (array-like) output instead'
)
# random_projection should not have elements too small,
# otherwise too much precision is lost in numerical gradient
def
random_projection
():
plain
=
rng
.
rand
(
*
o_fn_out
.
shape
)
+
0.5
if
cast_to_output_type
:
return
numpy
.
array
(
plain
,
o_output
.
dtype
)
return
plain
t_r
=
shared
(
random_projection
())
#random projection of o onto t_r
cost
=
sum
(
t_r
*
o_output
)
#This sum() is defined above, it's not the builtin sum.
cost_fn
=
function
(
tensor_pt
,
cost
)
#todo-- determine if this is actually needed
g_cost
=
as_tensor_variable
(
1.0
,
name
=
'g_cost'
)
if
cast_to_output_type
:
g_cost
=
cast
(
g_cost
,
o_output
.
dtype
)
symbolic_grad
=
grad
(
cost
,
tensor_pt
,
g_cost
,
disconnected_inputs
=
'ignore'
)
#if o_output.dtype in ['float32','float64']:
# assert all([x.dtype == o_output.dtype for x in symbolic_grad]),("Expected grad of type %s, got %s "%( symbolic_grad.dtype, o_output.dtyp))
grad_fn
=
function
(
tensor_pt
,
symbolic_grad
)
for
test_num
in
xrange
(
n_tests
):
num_grad
=
numeric_grad
(
cost_fn
,
[
p
.
copy
()
for
p
in
pt
],
eps
)
analytic_grad
=
grad_fn
(
*
[
p
.
copy
()
for
p
in
pt
])
if
not
isinstance
(
analytic_grad
,
(
list
,
tuple
)):
analytic_grad
=
[
analytic_grad
]
max_arg
,
max_err_pos
,
max_abs_err
,
max_rel_err
=
\
num_grad
.
max_err
(
analytic_grad
,
abs_tol
,
rel_tol
)
if
max_abs_err
>
abs_tol
and
max_rel_err
>
rel_tol
:
raise
verify_grad
.
E_grad
(
max_arg
,
max_err_pos
,
max_abs_err
,
max_rel_err
,
abs_tol
,
rel_tol
)
#get new random projection for next test
if
test_num
<
n_tests
-
1
:
t_r
.
set_value
(
random_projection
(),
borrow
=
True
)
class
GradientError
(
Exception
):
"""This error is raised when a gradient is calculated, but incorrect."""
def
__init__
(
self
,
arg
,
err_pos
,
abs_err
,
rel_err
,
abs_tol
,
rel_tol
):
self
.
arg
=
arg
self
.
err_pos
=
err_pos
self
.
abs_err
=
abs_err
self
.
rel_err
=
rel_err
self
.
abs_tol
=
abs_tol
self
.
rel_tol
=
rel_tol
def
__str__
(
self
):
return
"""GradientError: numeric gradient and analytic gradient exceed tolerance:
At position
%
i of argument
%
i,
abs. error =
%
f, abs. tolerance =
%
f
rel. error =
%
f, rel. tolerance =
%
f
"""
%
(
self
.
err_pos
,
self
.
arg
,
self
.
abs_err
,
self
.
abs_tol
,
self
.
rel_err
,
self
.
rel_tol
)
verify_grad
.
E_grad
=
GradientError
theano/tensor/elemwise.py
浏览文件 @
e4c6880e
...
@@ -218,6 +218,13 @@ class DimShuffle(Op):
...
@@ -218,6 +218,13 @@ class DimShuffle(Op):
rval
.
insert
(
augm
,
1
)
rval
.
insert
(
augm
,
1
)
return
[
rval
]
return
[
rval
]
def
R_op
(
self
,
inputs
,
eval_points
):
if
None
in
eval_points
:
return
[
None
]
return
self
.
make_node
(
*
eval_points
)
.
outputs
def
c_code
(
self
,
node
,
name
,
inp
,
out
,
sub
):
def
c_code
(
self
,
node
,
name
,
inp
,
out
,
sub
):
input
,
=
inp
input
,
=
inp
res
,
=
out
res
,
=
out
...
@@ -534,7 +541,78 @@ class Elemwise(Op):
...
@@ -534,7 +541,78 @@ class Elemwise(Op):
else
:
else
:
return
self
.
name
return
self
.
name
def
R_op
(
self
,
inputs
,
eval_points
):
outs
=
self
.
make_node
(
*
inputs
)
.
outputs
rval
=
[
None
for
x
in
outs
]
# For each output
for
idx
,
out
in
enumerate
(
outs
):
# make such that _bgrads computes only the gradients of the
# current output on the inputs ( and not all outputs)
ograds
=
[
theano
.
tensor
.
zeros_like
(
x
)
for
x
in
outs
]
ograds
[
idx
]
=
theano
.
tensor
.
ones_like
(
out
)
bgrads
=
self
.
_bgrad
(
inputs
,
ograds
)
rop_out
=
None
for
jdx
,
(
inp
,
eval_point
)
in
enumerate
(
zip
(
inputs
,
eval_points
)):
# if None, then we can just ignore this branch ..
# what we do is to assume that for any non-differentiable
# branch, the gradient is actually 0, which I think is not
# the right thing to do .. have to talk to Ian and James
# about it
if
bgrads
[
jdx
]
is
None
:
pass
elif
eval_point
is
not
None
:
if
rop_out
is
None
:
rop_out
=
bgrads
[
jdx
]
*
eval_point
else
:
rop_out
=
rop_out
+
bgrads
[
jdx
]
*
eval_point
rval
[
idx
]
=
rop_out
return
rval
def
grad
(
self
,
inputs
,
ograds
):
def
grad
(
self
,
inputs
,
ograds
):
#compute grad with respect to broadcasted input
rval
=
self
.
_bgrad
(
inputs
,
ograds
)
#sum out the broadcasted dimensions
for
i
,
ipt
in
enumerate
(
inputs
):
if
rval
[
i
]
is
None
:
continue
# list of all the dimensions that are broadcastable for input[i] so we
# can sum over them
# todo: only count dimensions that were effectively broadcasted
to_sum
=
[
j
for
j
,
bcast
in
enumerate
(
ipt
.
type
.
broadcastable
)
if
bcast
]
if
to_sum
:
shuffle
=
[]
j
=
0
for
bcast
in
ipt
.
type
.
broadcastable
:
if
bcast
==
1
:
shuffle
.
append
(
'x'
)
else
:
shuffle
.
append
(
j
)
j
+=
1
#close if
#close for
sr
=
Sum
(
axis
=
to_sum
)(
rval
[
i
])
sr
=
sr
.
dimshuffle
(
shuffle
)
#sr = DimShuffle(sr.type.broadcastable, shuffle)(sr)
rval
[
i
]
=
sr
#close if
#close for
return
rval
def
_bgrad
(
self
,
inputs
,
ograds
):
# returns grad, with respect to broadcasted versions of inputs
# Gradients (especially on the final costs) don't have to be symbolic
# Gradients (especially on the final costs) don't have to be symbolic
# e.g., ograds will be [ 1. ] if your objective is c and the output
# e.g., ograds will be [ 1. ] if your objective is c and the output
# of the current apply node is c
# of the current apply node is c
...
@@ -558,35 +636,17 @@ class Elemwise(Op):
...
@@ -558,35 +636,17 @@ class Elemwise(Op):
broadcastable
=
()),
broadcastable
=
()),
numpy
.
asarray
(
r
.
data
))
# .reshape(b)
numpy
.
asarray
(
r
.
data
))
# .reshape(b)
return
DimShuffle
((),
[
'x'
]
*
nd
,
inplace
=
True
)(
res
)
return
DimShuffle
((),
[
'x'
]
*
nd
,
inplace
=
True
)(
res
)
new_r
=
Elemwise
(
node
.
op
,
{})(
*
[
transform
(
i
nput
)
for
inpu
t
in
node
.
inputs
])
new_r
=
Elemwise
(
node
.
op
,
{})(
*
[
transform
(
i
pt
)
for
ip
t
in
node
.
inputs
])
return
new_r
return
new_r
ret
=
[]
ret
=
[]
for
scalar_igrad
,
i
npu
t
in
zip
(
scalar_igrads
,
inputs
):
for
scalar_igrad
,
i
p
t
in
zip
(
scalar_igrads
,
inputs
):
if
scalar_igrad
is
None
:
if
scalar_igrad
is
None
:
# undefined gradient
# undefined gradient
ret
.
append
(
None
)
ret
.
append
(
None
)
continue
continue
r
=
transform
(
scalar_igrad
)
r
et
.
append
(
transform
(
scalar_igrad
)
)
# list of all the dimensions that are broadcastable for that input so we
# can sum over them
# todo: only count dimensions that were effectively broadcasted
to_sum
=
[
i
for
i
,
bcast
in
enumerate
(
input
.
type
.
broadcastable
)
if
bcast
]
if
to_sum
:
shuffle
=
[]
j
=
0
for
bcast
in
input
.
type
.
broadcastable
:
if
bcast
==
1
:
shuffle
.
append
(
'x'
)
else
:
shuffle
.
append
(
j
)
j
+=
1
sr
=
Sum
(
axis
=
to_sum
)(
r
)
sr
=
DimShuffle
(
sr
.
type
.
broadcastable
,
shuffle
)(
sr
)
ret
.
append
(
sr
)
else
:
ret
.
append
(
r
)
return
ret
return
ret
def
perform
(
self
,
node
,
inputs
,
output_storage
):
def
perform
(
self
,
node
,
inputs
,
output_storage
):
...
@@ -1180,6 +1240,11 @@ class Sum(CAReduce):
...
@@ -1180,6 +1240,11 @@ class Sum(CAReduce):
i
+=
1
i
+=
1
return
Elemwise
(
scalar
.
second
)(
x
,
DimShuffle
(
gz
.
type
.
broadcastable
,
new_dims
)(
gz
)),
return
Elemwise
(
scalar
.
second
)(
x
,
DimShuffle
(
gz
.
type
.
broadcastable
,
new_dims
)(
gz
)),
def
R_op
(
self
,
inputs
,
eval_points
):
if
None
in
eval_points
:
return
[
None
]
return
self
.
make_node
(
*
eval_points
)
.
outputs
def
__str__
(
self
):
def
__str__
(
self
):
if
self
.
axis
is
None
:
if
self
.
axis
is
None
:
return
"Sum"
return
"Sum"
...
...
theano/tensor/nnet/nnet.py
浏览文件 @
e4c6880e
...
@@ -347,6 +347,13 @@ class Softmax(gof.Op):
...
@@ -347,6 +347,13 @@ class Softmax(gof.Op):
sm
=
softmax
(
x
)
sm
=
softmax
(
x
)
return
[
softmax_grad
(
g_sm
,
sm
)]
return
[
softmax_grad
(
g_sm
,
sm
)]
def
R_op
(
self
,
inputs
,
eval_points
):
# I think the Jacobian is symmetric so the R_op
# is the same as the grad
if
None
in
eval_points
:
return
[
None
]
return
self
.
grad
(
inputs
,
eval_points
)
def
infer_shape
(
self
,
node
,
shape
):
def
infer_shape
(
self
,
node
,
shape
):
return
shape
return
shape
...
...
theano/tensor/opt.py
浏览文件 @
e4c6880e
...
@@ -469,6 +469,11 @@ class MakeVector(T.Op):
...
@@ -469,6 +469,11 @@ class MakeVector(T.Op):
grads
.
append
(
output_gradients
[
0
][
i
])
grads
.
append
(
output_gradients
[
0
][
i
])
return
grads
return
grads
def
R_op
(
self
,
inputs
,
eval_points
):
if
None
in
eval_points
:
return
[
None
]
return
self
.
make_node
(
*
eval_points
)
.
outputs
make_vector
=
MakeVector
()
make_vector
=
MakeVector
()
class
MakeVectorPrinter
:
class
MakeVectorPrinter
:
...
...
theano/tensor/tensor_grad.py
0 → 100644
浏览文件 @
e4c6880e
"""Driver for gradient calculations."""
__authors__
=
"James Bergstra, Razvan Pascanu"
__copyright__
=
"(c) 2011, Universite de Montreal"
__license__
=
"3-clause BSD License"
__contact__
=
"theano-dev <theano-dev@googlegroups.com>"
__docformat__
=
"restructuredtext en"
import
__builtin__
import
logging
_logger
=
logging
.
getLogger
(
'theano.gradient'
)
import
sys
import
numpy
#for numeric_grad
import
theano
from
theano.tensor
import
TensorType
,
TensorVariable
,
ones_like
,
\
zeros_like
,
as_tensor_variable
from
theano
import
gradient
from
theano
import
gof
,
shared
from
theano
import
compile
########################
# R Operator
########################
def
Rop
(
f
,
wrt
,
eval_points
):
"""
Computes the R operation on `f` wrt to `wrt` evaluated at points given
in `eval_points`. Mathematically this stands for the jacobian of `f` wrt
to `wrt` right muliplied by the eval points.
:type f: `Variable` or list of `Variable`s
`f` stands for the output of the computational graph to which you
want to apply the R operator
:type wrt: `Variable` or list of `Variables`s
variables for which you compute the R operator of the expression
described by `f`
:type eval_points: `Variable` or list of `Variable`s
evalutation points for each of the variables in `wrt`
:rtype: `Variable` or list of `Variable`s depending on type of f
:return: symbolic expression such that
R_op[i] = sum_j ( d f[i] / d wrt[j]) eval_point[j]
where the indices in that expression are magic multidimensional
indices that specify both the position within a list and all
coordinates of the tensor element in the last
"""
if
not
isinstance
(
wrt
,
(
list
,
tuple
)):
wrt
=
[
wrt
]
if
not
isinstance
(
eval_points
,
(
list
,
tuple
)):
eval_points
=
[
eval_points
]
if
not
isinstance
(
f
,
(
list
,
tuple
)):
f
=
[
f
]
assert
len
(
wrt
)
==
len
(
eval_points
)
seen_nodes
=
{}
def
_traverse
(
node
):
if
node
is
None
:
return
None
else
:
op
=
node
.
op
inputs
=
node
.
inputs
if
not
hasattr
(
op
,
'R_op'
):
raise
Exception
((
' R_op was not implemented for
%
s'
' operation. Email the mailing list'
' for help'
)
%
op
.
__class__
.
__name__
)
# Compute the evaluation points corresponding to each of the
# inputs of the node
local_eval_points
=
[]
for
inp
in
inputs
:
if
inp
in
wrt
:
local_eval_points
.
append
(
eval_points
[
wrt
.
index
(
inp
)]
)
elif
inp
.
owner
is
None
:
local_eval_points
.
append
(
zeros_like
(
inp
)
)
elif
inp
.
owner
in
seen_nodes
:
local_eval_points
.
append
(
seen_nodes
[
inp
.
owner
][
inp
.
owner
.
outputs
.
index
(
inp
)
]
)
else
:
# We actually need to compute the R_op for this node
_traverse
(
inp
.
owner
)
local_eval_points
.
append
(
seen_nodes
[
inp
.
owner
][
inp
.
owner
.
outputs
.
index
(
inp
)
])
for
x
,
y
in
zip
(
inputs
,
local_eval_points
):
if
y
is
not
None
:
assert
(
as_tensor_variable
(
x
)
.
type
==
as_tensor_variable
(
y
)
.
type
)
seen_nodes
[
node
]
=
op
.
R_op
(
node
.
inputs
,
local_eval_points
)
return
None
# Populate the dictionary
for
out
in
f
:
_traverse
(
out
.
owner
)
rval
=
[]
for
out
in
f
:
if
out
in
wrt
:
rval
.
append
(
eval_points
[
wrt
.
index
(
out
)])
elif
seen_nodes
[
out
.
owner
][
out
.
owner
.
outputs
.
index
(
out
)]
is
None
:
raise
ValueError
((
'The function is not differentiable with '
'respect to the provided inputs !'
))
else
:
rval
.
append
(
seen_nodes
[
out
.
owner
][
out
.
owner
.
outputs
.
index
(
out
)]
)
if
len
(
rval
)
==
1
:
return
rval
[
0
]
else
:
return
rval
def
Lop
(
f
,
wrt
,
eval_points
,
consider_constant
=
[],
warn_type
=
False
,
disconnected_inputs
=
'raise'
):
"""
Computes the L operation on `f` wrt to `wrt` evaluated at points given
in `eval_points`. Mathematically this stands for the jacobian of `f` wrt
to `wrt` left muliplied by the eval points.
:type f: `Variable` or list of `Variable`s
`f` stands for the output of the computational graph to which you
want to apply the L operator
:type wrt: `Variable` or list of `Variables`s
variables for which you compute the L operator of the expression
described by `f`
:type eval_points: `Variable` or list of `Variable`s
evalutation points for each of the variables in `f`
:rtype: `Variable` or list of `Variable`s depending on type of f
:return: symbolic expression such that
L_op[i] = sum_i ( d f[i] / d wrt[j]) eval_point[i]
where the indices in that expression are magic multidimensional
indices that specify both the position within a list and all
coordinates of the tensor element in the last
"""
if
not
isinstance
(
f
,
TensorVariable
):
raise
TypeError
(
'In tensor.Lop(), cost argument should be a TensorVariable.'
,
f
)
if
type
(
eval_points
)
not
in
(
list
,
tuple
):
eval_points
=
[
eval_points
]
if
type
(
f
)
not
in
(
list
,
tuple
):
f
=
[
f
]
inputs
=
gof
.
graph
.
inputs
(
f
)
gmap
=
gradient
.
grad_sources_inputs
(
zip
(
f
,
eval_points
),
list
(
inputs
)
+
list
(
consider_constant
),
warn_type
=
warn_type
)
# Note : If p is not in gmap there can be several reasons, among which
# is the fact that p might not be part of the computational graph. A
# simple example is that for a+b for e.g. a[0] is not part of the graph,
# so Theano does not know how to compute TT.grad(TT.sum(a+b), a[0])
# such subtle cases can be fixed by a more careful implementation of the
# gradient, but for now Theano needs to throw an exception, and make the
# user aware that it does not know how to compute that gradient
if
not
isinstance
(
wrt
,
(
list
,
tuple
)):
wrt
=
[
wrt
]
ret
=
[]
for
p
in
wrt
:
if
p
in
gmap
:
ret
.
append
(
gmap
[
p
])
else
:
message
=
(
"Lop method was asked to compute the gradient "
"with respect to a variable that is not part of "
"the computational graph of the cost, or is used "
"only by a non-differentiable operator:
%
s"
%
p
)
if
disconnected_inputs
==
'ignore'
:
pass
elif
disconnected_inputs
==
'warn'
:
warnings
.
warn
(
message
,
stacklevel
=
1
)
elif
disconnected_inputs
==
'raise'
:
raise
ValueError
(
message
)
else
:
raise
ValueError
(
"Invalid value for keyword "
"'disconnected_inputs', valid values are "
"'ignore', 'warn' and 'raise'."
)
ret
.
append
(
zeros_like
(
p
))
if
len
(
ret
)
==
1
:
return
ret
[
0
]
else
:
return
ret
#########################
# Gradient
#########################
def
grad
(
cost
,
wrt
,
g_cost
=
None
,
consider_constant
=
[],
warn_type
=
False
,
disconnected_inputs
=
'raise'
):
"""
:type cost: Scalar (0-dimensional) `Variable`
:type wrt: `Variable` or list of `Variable`s.
:type g_cost: Scalar `Variable`, or None
:param g_cost: an expression for the gradient through cost. The default is
``ones_like(cost)``.
:param consider_constant: a list of expressions not to backpropagate through
:param warn_type: a value of True will cause warnings to be logged for any Op that emits a
gradient that does not match its input type.
:type disconnected_inputs: string
:param disconnected_inputs: Defines the behaviour if some of the variables
in ``wrt`` are not part of the computational graph computing ``cost``
(or if all links are non-differentiable). The possible values are:
- 'ignore': considers that the gradient on these parameters is zero.
- 'warn': consider the gradient zero, and print a warning.
- 'raise': raise an exception.
:rtype: `Variable` or list of `Variable`s (depending upon `wrt`)
:return: symbolic expression of gradient of `cost` with respect to `wrt`.
If `wrt` is a list, then return a list containing the gradient of `cost` wrt
each element of the list. If an element of `wrt` is not differentiable
with respect to the output, then a zero variable is returned.
This function is a wrapper around the more general function
`theano.gradient.grad_sources_inputs``.
"""
if
not
isinstance
(
cost
,
TensorVariable
):
raise
TypeError
(
'In tensor.grad(), cost argument should be a TensorVariable.'
,
cost
)
if
cost
.
type
.
ndim
:
raise
TypeError
(
'In tensor.grad, "cost" argument should be a scalar, but ndim'
' is
%
i (should be 0). If you want to compute the gradient of'
' the sum of cost, you should use cost.sum().'
%
cost
.
type
.
ndim
)
if
g_cost
is
None
:
g_cost
=
ones_like
(
cost
)
inputs
=
gof
.
graph
.
inputs
([
cost
])
gmap
=
gradient
.
grad_sources_inputs
(
[(
cost
,
g_cost
)],
list
(
inputs
)
+
list
(
consider_constant
),
warn_type
=
warn_type
)
# Note : If p is not in gmap there can be several reasons, among which
# is the fact that p might not be part of the computational graph. A
# simple example is that for a+b for e.g. a[0] is not part of the graph,
# so Theano does not know how to compute TT.grad(TT.sum(a+b), a[0])
# such subtle cases can be fixed by a more careful implementation of the
# gradient, but for now Theano needs to throw an exception, and make the
# user aware that it does not know how to compute that gradient
if
not
isinstance
(
wrt
,
(
list
,
tuple
)):
wrt
=
[
wrt
]
ret
=
[]
for
p
in
wrt
:
if
p
in
gmap
:
ret
.
append
(
gmap
[
p
])
else
:
message
=
(
"grad method was asked to compute the gradient "
"with respect to a variable that is not part of "
"the computational graph of the cost, or is used "
"only by a non-differentiable operator:
%
s"
%
p
)
if
disconnected_inputs
==
'ignore'
:
pass
elif
disconnected_inputs
==
'warn'
:
warnings
.
warn
(
message
,
stacklevel
=
1
)
elif
disconnected_inputs
==
'raise'
:
raise
ValueError
(
message
)
else
:
raise
ValueError
(
"Invalid value for keyword "
"'disconnected_inputs', valid values are "
"'ignore', 'warn' and 'raise'."
)
ret
.
append
(
zeros_like
(
p
))
if
len
(
ret
)
==
1
:
return
ret
[
0
]
else
:
return
ret
class
numeric_grad
:
"""WRITEME"""
# Note on step sizes and tolerances:
#
# There is a relationship between the step size and the function value and the measurement
# error that is incurred due to rounding. The finite difference we measure is
# delta = f(x0) - f(x0+eps)
#
# For maximum precision, f should be close to zero.
# For every power of 2 that f departs from zero, we lose a bit of precision in delta.
#
# Even in this case of maximum accuracy, there is a tradeoff between stepsize and
# measurement error.
# Taking small steps allows us to measure large derivatives accuractly, but longer steps
# are required to measure small derivatives accurately. However longer steps introduce
# bias into our measurement in general for non-linear functions.
#
# It would be interesting to have a version of numeric grad that used an adaptive stepsize.
#
# For now, we use a heuristic that catches very bad gradients, but is not perfectly
# accurate.
type_eps
=
{
'float64'
:
1e-7
,
'float32'
:
3e-4
,
numpy
.
dtype
(
'float64'
):
1e-7
,
numpy
.
dtype
(
'float32'
):
3e-4
}
def
__init__
(
self
,
f
,
pt
,
eps
=
None
):
"""Return the gradient of f at pt.
This function computes the gradient by a one-sided finite differences of a
fixed step size (eps).
It is assumed that f(...) will return a scalar.
It is assumed that all f's inputs are numpy.ndarray objects.
:param eps: the stepsize for the finite differencing. None means input
dtype-dependent. See `type_eps`.
"""
def
prod
(
inputs
):
rval
=
1
for
i
in
inputs
:
rval
*=
i
return
rval
packed_pt
=
False
if
not
isinstance
(
pt
,
(
list
,
tuple
)):
pt
=
[
pt
]
packed_pt
=
True
apt
=
[
numpy
.
array
(
p
)
for
p
in
pt
]
shapes
=
[
p
.
shape
for
p
in
apt
]
dtypes
=
[
str
(
p
.
dtype
)
for
p
in
apt
]
# TODO: remove this eventually (why was this here in the first place ?)
# In the case of CSM, the arguments are a mixture of floats and integers...
#if not dtypes == [dtypes[0]] * len(apt):
#raise TypeError('All function arguments must have same dtype')
total_size
=
__builtin__
.
sum
(
prod
(
sh
)
for
sh
in
shapes
)
working_dtype
=
__builtin__
.
min
((
self
.
type_eps
[
dt
],
dt
)
for
dt
in
dtypes
)[
1
]
#create un-initialized memory
x
=
numpy
.
ndarray
((
total_size
,),
dtype
=
working_dtype
)
gx
=
numpy
.
ndarray
((
total_size
,),
dtype
=
working_dtype
)
if
eps
is
None
:
eps
=
__builtin__
.
max
(
self
.
type_eps
[
dt
]
for
dt
in
dtypes
)
#set up aliases so that apt[i] is backed by memory in x
# and self.gf is backed by memory in gx
cur_pos
=
0
self
.
gf
=
[]
for
i
,
p
in
enumerate
(
apt
):
p_size
=
prod
(
p
.
shape
)
# set up alias
apt
[
i
]
=
x
[
cur_pos
:
cur_pos
+
p_size
]
.
reshape
(
p
.
shape
)
self
.
gf
.
append
(
gx
[
cur_pos
:
cur_pos
+
p_size
]
.
reshape
(
p
.
shape
))
# initialize with p's value
apt
[
i
][
...
]
=
p
cur_pos
+=
p_size
f_x
=
f
(
*
[
p
.
copy
()
for
p
in
apt
])
# now iterate over the elements of x, and call f on apt.
x_copy
=
x
.
copy
()
for
i
in
xrange
(
total_size
):
x
[:]
=
x_copy
x
[
i
]
+=
eps
f_eps
=
f
(
*
apt
)
gx
[
i
]
=
numpy
.
asarray
((
f_eps
-
f_x
)
/
eps
)
if
packed_pt
:
self
.
gf
=
self
.
gf
[
0
]
@staticmethod
def
abs_rel_err
(
a
,
b
):
"""Return absolute and relative error between a and b.
The relative error is a small number when a and b are close, relative to how big they are.
Formulas used:
abs_err = abs(a - b)
rel_err = abs_err / (abs(a) + abs(b))
The tuple (abs_err, rel_err) is returned
"""
abs_err
=
abs
(
a
-
b
)
rel_err
=
abs_err
/
(
abs
(
a
)
+
abs
(
b
))
return
(
abs_err
,
rel_err
)
def
abs_rel_errors
(
self
,
g_pt
):
"""Return the abs and rel error of gradient estimate `g_pt`
`g_pt` must be a list of ndarrays of the same length as self.gf, otherwise a ValueError
is raised.
Corresponding ndarrays in `g_pt` and `self.gf` must have the same shape or ValueError
is raised.
"""
if
len
(
g_pt
)
!=
len
(
self
.
gf
):
raise
ValueError
(
'argument has wrong number of elements'
,
len
(
g_pt
))
errs
=
[]
for
i
,
(
a
,
b
)
in
enumerate
(
zip
(
g_pt
,
self
.
gf
)):
if
a
.
shape
!=
b
.
shape
:
raise
ValueError
(
'argument element
%
i has wrong shape
%
s'
%
(
i
,
str
((
a
.
shape
,
b
.
shape
))))
errs
.
append
(
numeric_grad
.
abs_rel_err
(
a
,
b
))
return
errs
def
max_err
(
self
,
g_pt
,
abs_tol
,
rel_tol
):
"""Find the biggest error between g_pt and self.gf.
What is measured is the violation of relative and absolute errors,
wrt the provided tolerances (abs_tol, rel_tol).
A value > 1 means both tolerances are exceeded.
Return the argmax of min(abs_err / abs_tol, rel_err / rel_tol) over g_pt,
as well as abs_err and rel_err at this point.
"""
pos
=
[]
errs
=
[]
abs_errs
=
[]
rel_errs
=
[]
abs_rel_errs
=
self
.
abs_rel_errors
(
g_pt
)
for
abs_err
,
rel_err
in
abs_rel_errs
:
scaled_err
=
numpy
.
minimum
(
abs_err
/
abs_tol
,
rel_err
/
rel_tol
)
max_i
=
scaled_err
.
argmax
()
pos
.
append
(
max_i
)
errs
.
append
(
scaled_err
.
flatten
()[
max_i
])
abs_errs
.
append
(
abs_err
.
flatten
()[
max_i
])
rel_errs
.
append
(
rel_err
.
flatten
()[
max_i
])
# max over the arrays in g_pt
max_arg
=
numpy
.
argmax
(
errs
)
max_pos
=
pos
[
max_arg
]
return
(
max_arg
,
pos
[
max_arg
],
abs_errs
[
max_arg
],
rel_errs
[
max_arg
])
def
verify_grad
(
fun
,
pt
,
n_tests
=
2
,
rng
=
None
,
eps
=
None
,
abs_tol
=
None
,
rel_tol
=
None
,
mode
=
None
,
cast_to_output_type
=
False
):
""" Test a gradient by Finite Difference Method. Raise error on failure.
Example:
>>> verify_grad(theano.tensor.tanh,
(numpy.asarray([[2,3,4], [-1, 3.3, 9.9]]),),
rng=numpy.random)
Raises an Exception if the difference between the analytic gradient and
numerical gradient (computed through the Finite Difference Method) of a random
projection of the fun's output to a scalar exceeds
the given tolerance.
:param fun: a Python function that takes Theano variables as inputs,
and returns a Theano variable. For instance, an Op instance with
a single output.
:param pt: the list of numpy.ndarrays to use as input values.
These arrays must be either float32 or float64 arrays.
:param n_tests: number of times to run the test
:param rng: random number generator used to sample u, we test gradient of sum(u * fun) at pt
:param eps: stepsize used in the Finite Difference Method (Default None is type-dependent)
:param abs_tol: absolute tolerance used as threshold for gradient comparison
:param rel_tol: relative tolerance used as threshold for gradient comparison
:note: WARNING to unit-test writers: if `op` is a function that builds a graph,
try to make it a SMALL graph. Often verify grad is run in
debug mode, which can be very slow if it has to verify a lot
of intermediate computations.
:note: This op does not support multiple outputs. In tests/test_scan.py there is
an experimental verify_grad that covers that case as well by using random
projections.
"""
assert
isinstance
(
pt
,
(
list
,
tuple
))
pt
=
[
numpy
.
array
(
p
)
for
p
in
pt
]
for
i
,
p
in
enumerate
(
pt
):
if
p
.
dtype
not
in
(
'float32'
,
'float64'
):
raise
TypeError
((
'verify_grad can work only with floating point '
'inputs, but input
%
i has dtype "
%
s".'
)
%
(
i
,
p
.
dtype
))
_type_tol
=
dict
(
# relativ error tolerances for different types
float32
=
1e-2
,
float64
=
1e-4
)
if
abs_tol
is
None
:
abs_tol
=
__builtin__
.
max
(
_type_tol
[
str
(
p
.
dtype
)]
for
p
in
pt
)
if
rel_tol
is
None
:
rel_tol
=
__builtin__
.
max
(
_type_tol
[
str
(
p
.
dtype
)]
for
p
in
pt
)
if
rng
is
None
:
raise
TypeError
(
'rng should be a valid instance of numpy.random.RandomState.'
,
'You may want to use theano.tests.unittest_tools.verify_grad instead of theano.tensor.verify_grad.'
)
# We allow input downcast in function, because numeric_grad works in the
# most precise dtype used among the inputs, so we may need to cast some.
def
function
(
inputs
,
output
):
if
mode
is
None
:
f
=
compile
.
function
(
inputs
,
output
,
accept_inplace
=
True
,
allow_input_downcast
=
True
)
else
:
f
=
compile
.
function
(
inputs
,
output
,
accept_inplace
=
True
,
allow_input_downcast
=
True
,
mode
=
mode
)
return
f
tensor_pt
=
[
TensorType
(
as_tensor_variable
(
p
)
.
dtype
,
as_tensor_variable
(
p
)
.
broadcastable
)(
name
=
'input
%
i'
%
i
)
for
i
,
p
in
enumerate
(
pt
)]
#fun can be either a function or an actual Op instance
o_output
=
fun
(
*
tensor_pt
)
if
isinstance
(
o_output
,
list
):
raise
NotImplementedError
(
'cant (yet) autotest gradient of fun with multiple outputs'
)
# we could make loop over outputs making random projections R for each,
# but this doesn't handle the case where not all the outputs are
# differentiable... so I leave this as TODO for now -JB.
o_fn
=
function
(
tensor_pt
,
o_output
)
o_fn_out
=
o_fn
(
*
[
p
.
copy
()
for
p
in
pt
])
if
isinstance
(
o_fn_out
,
tuple
)
or
isinstance
(
o_fn_out
,
list
):
raise
TypeError
(
'It seems like you are trying to use verify_grad '
'on an op or a function which outputs a list: there should'
' be a single (array-like) output instead'
)
# random_projection should not have elements too small,
# otherwise too much precision is lost in numerical gradient
def
random_projection
():
plain
=
rng
.
rand
(
*
o_fn_out
.
shape
)
+
0.5
if
cast_to_output_type
:
return
numpy
.
array
(
plain
,
o_output
.
dtype
)
return
plain
t_r
=
shared
(
random_projection
())
#random projection of o onto t_r
cost
=
theano
.
tensor
.
sum
(
t_r
*
o_output
)
#This sum() is defined above, it's not the builtin sum.
cost_fn
=
function
(
tensor_pt
,
cost
)
#todo-- determine if this is actually needed
g_cost
=
as_tensor_variable
(
1.0
,
name
=
'g_cost'
)
if
cast_to_output_type
:
g_cost
=
cast
(
g_cost
,
o_output
.
dtype
)
symbolic_grad
=
grad
(
cost
,
tensor_pt
,
g_cost
,
disconnected_inputs
=
'ignore'
)
#if o_output.dtype in ['float32','float64']:
# assert all([x.dtype == o_output.dtype for x in symbolic_grad]),("Expected grad of type %s, got %s "%( symbolic_grad.dtype, o_output.dtyp))
grad_fn
=
function
(
tensor_pt
,
symbolic_grad
)
for
test_num
in
xrange
(
n_tests
):
num_grad
=
numeric_grad
(
cost_fn
,
[
p
.
copy
()
for
p
in
pt
],
eps
)
analytic_grad
=
grad_fn
(
*
[
p
.
copy
()
for
p
in
pt
])
if
not
isinstance
(
analytic_grad
,
(
list
,
tuple
)):
analytic_grad
=
[
analytic_grad
]
max_arg
,
max_err_pos
,
max_abs_err
,
max_rel_err
=
\
num_grad
.
max_err
(
analytic_grad
,
abs_tol
,
rel_tol
)
if
max_abs_err
>
abs_tol
and
max_rel_err
>
rel_tol
:
raise
verify_grad
.
E_grad
(
max_arg
,
max_err_pos
,
max_abs_err
,
max_rel_err
,
abs_tol
,
rel_tol
)
#get new random projection for next test
if
test_num
<
n_tests
-
1
:
t_r
.
set_value
(
random_projection
(),
borrow
=
True
)
class
GradientError
(
Exception
):
"""This error is raised when a gradient is calculated, but incorrect."""
def
__init__
(
self
,
arg
,
err_pos
,
abs_err
,
rel_err
,
abs_tol
,
rel_tol
):
self
.
arg
=
arg
self
.
err_pos
=
err_pos
self
.
abs_err
=
abs_err
self
.
rel_err
=
rel_err
self
.
abs_tol
=
abs_tol
self
.
rel_tol
=
rel_tol
def
__str__
(
self
):
return
"""GradientError: numeric gradient and analytic gradient exceed tolerance:
At position
%
i of argument
%
i,
abs. error =
%
f, abs. tolerance =
%
f
rel. error =
%
f, rel. tolerance =
%
f
"""
%
(
self
.
err_pos
,
self
.
arg
,
self
.
abs_err
,
self
.
abs_tol
,
self
.
rel_err
,
self
.
rel_tol
)
verify_grad
.
E_grad
=
GradientError
theano/tensor/tests/test_rop.py
浏览文件 @
e4c6880e
...
@@ -2,6 +2,48 @@
...
@@ -2,6 +2,48 @@
WRITE ME
WRITE ME
Tests for the R operator / L operator
Tests for the R operator / L operator
ops without:
PermuteRowElements
Tile
AdvancedSubtensor
TensorDot
Outer
Prod
MulwithoutZeros
ProdWithoutZeros
list of ops that support R-op:
* Alloc
* Split
* ARange
* ScalarFromTensor
* Shape
* SpecifyShape
* MaxAndArgmax
* Subtensor
* IncSubtensor
* Rebroadcast
* Join
* Reshape
* Flatten
* AdvancedSubtensor1
* AdvancedIncSubtensor1
* AdvancedIncSubtensor
* Dot
* DimShuffle
* Elemwise
* Sum
* Softmax
* Scan
"""
"""
import
unittest
import
unittest
...
@@ -10,52 +52,208 @@ from theano import function
...
@@ -10,52 +52,208 @@ from theano import function
import
theano
import
theano
import
theano.tensor
as
TT
import
theano.tensor
as
TT
import
numpy
import
numpy
from
theano.gof
import
Op
,
Apply
class
test_rop
(
unittest
.
TestCase
):
'''
Special Op created to test what happens when you have one op that is not
differentiable in the computational graph
'''
class
BreakRop
(
Op
):
"""
@note: Non-differentiable.
"""
def
__hash__
(
self
):
return
hash
(
type
(
self
))
def
__eq__
(
self
,
other
):
return
type
(
self
)
==
type
(
other
)
def
make_node
(
self
,
x
):
return
Apply
(
self
,
[
x
],
[
x
.
type
()])
def
perform
(
self
,
node
,
inp
,
out_
):
x
,
=
inp
out
,
=
out_
out
[
0
]
=
x
def
grad
(
self
,
inp
,
grads
):
return
[
None
]
def
R_op
(
self
,
inputs
,
eval_points
):
return
[
None
]
def
test_specifyshape
(
self
):
break_op
=
BreakRop
()
rng
=
numpy
.
random
.
RandomState
(
utt
.
fetch_seed
())
vx
=
numpy
.
asarray
(
rng
.
uniform
(
size
=
(
5
,)),
theano
.
config
.
floatX
)
vv
=
numpy
.
asarray
(
rng
.
uniform
(
size
=
(
5
,)),
theano
.
config
.
floatX
)
class
test_RopLop
(
unittest
.
TestCase
):
x
=
TT
.
vector
(
'x'
)
def
setUp
(
self
):
v
=
TT
.
vector
(
'v'
)
# Using vectors make things a lot simpler for generating the same
y
=
TT
.
specify_shape
(
x
,
(
5
,))
# computations using scan
yv
=
TT
.
Rop
(
y
,
x
,
v
)
self
.
x
=
TT
.
vector
(
'x'
)
rop_f
=
function
([
x
,
v
],
yv
)
self
.
v
=
TT
.
vector
(
'v'
)
J
,
_
=
theano
.
scan
(
lambda
i
,
y
,
x
:
TT
.
grad
(
y
[
i
],
x
),
self
.
rng
=
numpy
.
random
.
RandomState
(
utt
.
fetch_seed
())
sequences
=
TT
.
arange
(
x
.
shape
[
0
]),
self
.
in_shape
=
(
5
+
self
.
rng
.
randint
(
30
),)
non_sequences
=
[
y
,
x
])
self
.
mx
=
TT
.
matrix
(
'mx'
)
sy
=
TT
.
dot
(
J
,
v
)
self
.
mv
=
TT
.
matrix
(
'mv'
)
self
.
mat_in_shape
=
(
5
+
self
.
rng
.
randint
(
30
),
5
+
self
.
rng
.
randint
(
30
))
def
check_nondiff_rop
(
self
,
y
):
raised
=
False
try
:
tmp
=
TT
.
Rop
(
y
,
self
.
x
,
self
.
v
)
except
ValueError
:
raised
=
True
if
not
raised
:
self
.
fail
((
'Op did not raised an error even though the function'
' is not differentiable'
))
def
check_mat_rop_lop
(
self
,
y
,
out_shape
):
vx
=
numpy
.
asarray
(
self
.
rng
.
uniform
(
size
=
self
.
mat_in_shape
),
theano
.
config
.
floatX
)
vv
=
numpy
.
asarray
(
self
.
rng
.
uniform
(
size
=
self
.
mat_in_shape
),
theano
.
config
.
floatX
)
yv
=
TT
.
Rop
(
y
,
self
.
mx
,
self
.
mv
)
rop_f
=
function
([
self
.
mx
,
self
.
mv
],
yv
)
sy
,
_
=
theano
.
scan
(
lambda
i
,
y
,
x
,
v
:
(
TT
.
grad
(
y
[
i
],
x
)
*
v
)
.
sum
(),
sequences
=
TT
.
arange
(
y
.
shape
[
0
]),
non_sequences
=
[
y
,
self
.
mx
,
self
.
mv
])
scan_f
=
function
([
self
.
mx
,
self
.
mv
],
sy
)
scan_f
=
function
([
x
,
v
],
sy
)
v1
=
rop_f
(
vx
,
vv
)
v1
=
rop_f
(
vx
,
vv
)
v2
=
scan_f
(
vx
,
vv
)
v2
=
scan_f
(
vx
,
vv
)
assert
numpy
.
allclose
(
v1
,
v2
)
assert
numpy
.
allclose
(
v1
,
v2
)
self
.
check_nondiff_rop
(
theano
.
clone
(
y
,
replace
=
{
self
.
mx
:
break_op
(
self
.
mx
)}))
vv
=
numpy
.
asarray
(
self
.
rng
.
uniform
(
size
=
out_shape
),
theano
.
config
.
floatX
)
yv
=
TT
.
Lop
(
y
,
self
.
mx
,
self
.
v
)
lop_f
=
function
([
self
.
mx
,
self
.
v
],
yv
)
sy
,
_
=
theano
.
scan
(
lambda
i
,
y
,
x
,
v
:
(
TT
.
grad
(
y
[
i
]
*
v
[
i
],
x
))[
i
],
sequences
=
TT
.
arange
(
y
.
shape
[
0
]),
non_sequences
=
[
y
,
self
.
mx
,
self
.
v
])
scan_f
=
function
([
self
.
mx
,
self
.
v
],
sy
)
class
test_lop
(
unittest
.
TestCase
):
def
test_specifyshape
(
self
):
v1
=
lop_f
(
vx
,
vv
)
rng
=
numpy
.
random
.
RandomState
(
utt
.
fetch_seed
())
v2
=
scan_f
(
vx
,
vv
)
vx
=
numpy
.
asarray
(
rng
.
uniform
(
size
=
(
5
,)),
theano
.
config
.
floatX
)
assert
numpy
.
allclose
(
v1
,
v2
)
vv
=
numpy
.
asarray
(
rng
.
uniform
(
size
=
(
5
,)),
theano
.
config
.
floatX
)
x
=
TT
.
vector
(
'x'
)
def
check_rop_lop
(
self
,
y
,
out_shape
):
v
=
TT
.
vector
(
'v'
)
# TEST ROP
y
=
TT
.
specify_shape
(
x
,
(
5
,))
vx
=
numpy
.
asarray
(
self
.
rng
.
uniform
(
size
=
self
.
in_shape
),
theano
.
config
.
floatX
)
yv
=
TT
.
Lop
(
y
,
x
,
v
)
vv
=
numpy
.
asarray
(
self
.
rng
.
uniform
(
size
=
self
.
in_shape
),
theano
.
config
.
floatX
)
rop_f
=
function
([
x
,
v
],
yv
)
yv
=
TT
.
Rop
(
y
,
self
.
x
,
self
.
v
)
rop_f
=
function
([
self
.
x
,
self
.
v
],
yv
)
J
,
_
=
theano
.
scan
(
lambda
i
,
y
,
x
:
TT
.
grad
(
y
[
i
],
x
),
J
,
_
=
theano
.
scan
(
lambda
i
,
y
,
x
:
TT
.
grad
(
y
[
i
],
x
),
sequences
=
TT
.
arange
(
x
.
shape
[
0
]),
sequences
=
TT
.
arange
(
y
.
shape
[
0
]),
non_sequences
=
[
y
,
x
])
non_sequences
=
[
y
,
self
.
x
])
sy
=
TT
.
dot
(
v
,
J
)
sy
=
TT
.
dot
(
J
,
self
.
v
)
scan_f
=
function
([
x
,
v
],
sy
)
scan_f
=
function
([
self
.
x
,
self
.
v
],
sy
)
v1
=
rop_f
(
vx
,
vv
)
v1
=
rop_f
(
vx
,
vv
)
v2
=
scan_f
(
vx
,
vv
)
v2
=
scan_f
(
vx
,
vv
)
assert
numpy
.
allclose
(
v1
,
v2
)
assert
numpy
.
allclose
(
v1
,
v2
)
self
.
check_nondiff_rop
(
theano
.
clone
(
y
,
replace
=
{
self
.
x
:
break_op
(
self
.
x
)}))
# TEST LOP
vx
=
numpy
.
asarray
(
self
.
rng
.
uniform
(
size
=
self
.
in_shape
),
theano
.
config
.
floatX
)
vv
=
numpy
.
asarray
(
self
.
rng
.
uniform
(
size
=
out_shape
),
theano
.
config
.
floatX
)
yv
=
TT
.
Lop
(
y
,
self
.
x
,
self
.
v
)
lop_f
=
function
([
self
.
x
,
self
.
v
],
yv
)
J
,
_
=
theano
.
scan
(
lambda
i
,
y
,
x
:
TT
.
grad
(
y
[
i
],
x
),
sequences
=
TT
.
arange
(
y
.
shape
[
0
]),
non_sequences
=
[
y
,
self
.
x
])
sy
=
TT
.
dot
(
self
.
v
,
J
)
scan_f
=
function
([
self
.
x
,
self
.
v
],
sy
)
v1
=
lop_f
(
vx
,
vv
)
v2
=
scan_f
(
vx
,
vv
)
assert
numpy
.
allclose
(
v1
,
v2
)
def
test_shape
(
self
):
self
.
check_nondiff_rop
(
self
.
x
.
shape
[
0
])
def
test_specifyshape
(
self
):
self
.
check_rop_lop
(
TT
.
specify_shape
(
self
.
x
,
self
.
in_shape
),
self
.
in_shape
)
def
test_max_argmax
(
self
):
self
.
check_map_rop_lop
(
TT
.
max
(
self
.
mx
,
axis
=
1
),
(
self
.
mat_in_shape
[
0
],))
def
test_max_argmax
(
self
):
self
.
check_nondiff_rop
(
TT
.
argmax
(
self
.
mx
,
axis
=
1
))
def
test_subtensor
(
self
):
self
.
check_rop_lop
(
self
.
x
[:
4
],
(
4
,))
def
test_incsubtensor1
(
self
):
tv
=
numpy
.
asarray
(
self
.
rng
.
uniform
(
size
=
(
3
,)),
theano
.
config
.
floatX
)
t
=
theano
.
shared
(
tv
)
out
=
TT
.
inc_subtensor
(
self
.
x
[:
3
],
t
)
self
.
check_rop_lop
(
out
,
self
.
in_shape
)
def
test_incsubtensor1
(
self
):
tv
=
numpy
.
asarray
(
self
.
rng
.
uniform
(
size
=
(
10
,)),
theano
.
config
.
floatX
)
t
=
theano
.
shared
(
tv
)
out
=
TT
.
inc_subtensor
(
t
[:
4
],
self
.
x
[:
4
])
self
.
check_rop_lop
(
out
,
(
10
,))
def
test_setsubtensor1
(
self
):
tv
=
numpy
.
asarray
(
self
.
rng
.
uniform
(
size
=
(
3
,)),
theano
.
config
.
floatX
)
t
=
theano
.
shared
(
tv
)
out
=
TT
.
set_subtensor
(
self
.
x
[:
3
],
t
)
self
.
check_rop_lop
(
out
,
self
.
in_shape
)
def
test_setsubtensor1
(
self
):
tv
=
numpy
.
asarray
(
self
.
rng
.
uniform
(
size
=
(
10
,)),
theano
.
config
.
floatX
)
t
=
theano
.
shared
(
tv
)
out
=
TT
.
set_subtensor
(
t
[:
4
],
self
.
x
[:
4
])
self
.
check_rop_lop
(
out
,
(
10
,))
def
test_join
(
self
):
tv
=
numpy
.
asarray
(
self
.
rng
.
uniform
(
size
=
(
10
,)),
theano
.
config
.
floatX
)
t
=
theano
.
shared
(
tv
)
out
=
TT
.
join
(
0
,
self
.
x
,
t
)
self
.
check_rop_lop
(
out
,
(
self
.
in_shape
[
0
]
+
10
,))
def
test_dot
(
self
):
insh
=
self
.
in_shape
[
0
]
vW
=
numpy
.
asarray
(
self
.
rng
.
uniform
(
size
=
(
insh
,
insh
)),
theano
.
config
.
floatX
)
W
=
theano
.
shared
(
vW
)
self
.
check_rop_lop
(
TT
.
dot
(
self
.
x
,
W
),
self
.
in_shape
)
def
test_elemwise0
(
self
):
self
.
check_rop_lop
(
(
self
.
x
+
1
)
**
2
,
self
.
in_shape
)
def
test_elemwise1
(
self
):
self
.
check_rop_lop
(
self
.
x
+
TT
.
cast
(
self
.
x
,
'int32'
),
self
.
in_shape
)
def
test_sum
(
self
):
self
.
check_mat_rop_lop
(
self
.
mx
.
sum
(
axis
=
1
),
(
self
.
mat_in_shape
[
0
],))
def
test_softmax
(
self
):
# Softmax adds an extra dimnesion !
self
.
check_rop_lop
(
TT
.
nnet
.
softmax
(
self
.
x
)[
0
],
self
.
in_shape
)
编写
预览
Markdown
格式
0%
重试
或
添加新文件
添加附件
取消
您添加了
0
人
到此讨论。请谨慎行事。
请先完成此评论的编辑!
取消
请
注册
或者
登录
后发表评论