Skip to content
项目
群组
代码片段
帮助
当前项目
正在载入...
登录 / 注册
切换导航面板
P
pytensor
项目
项目
详情
活动
周期分析
仓库
仓库
文件
提交
分支
标签
贡献者
图表
比较
统计图
议题
0
议题
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
CI / CD
CI / CD
流水线
作业
日程
统计图
Wiki
Wiki
代码片段
代码片段
成员
成员
折叠边栏
关闭边栏
活动
图像
聊天
创建新问题
作业
提交
问题看板
Open sidebar
testgroup
pytensor
Commits
52cb8ec7
提交
52cb8ec7
authored
11月 07, 2014
作者:
Frédéric Bastien
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #2228 from lamblin/fix_float16
Prevent computations in float16 in scalar and elemwise
上级
d7071622
81369296
显示空白字符变更
内嵌
并排
正在显示
7 个修改的文件
包含
539 行增加
和
176 行删除
+539
-176
basic.py
theano/scalar/basic.py
+121
-2
test_basic.py
theano/scalar/tests/test_basic.py
+130
-3
basic.py
theano/tensor/basic.py
+5
-5
elemwise.py
theano/tensor/elemwise.py
+23
-44
sigm.py
theano/tensor/nnet/sigm.py
+15
-2
test_sigm.py
theano/tensor/nnet/tests/test_sigm.py
+12
-11
test_basic.py
theano/tensor/tests/test_basic.py
+233
-109
没有找到文件。
theano/scalar/basic.py
浏览文件 @
52cb8ec7
...
@@ -1504,7 +1504,7 @@ class TrueDiv(BinaryScalarOp):
...
@@ -1504,7 +1504,7 @@ class TrueDiv(BinaryScalarOp):
x
=
numpy
.
asarray
(
x
)
x
=
numpy
.
asarray
(
x
)
y
=
numpy
.
asarray
(
y
)
y
=
numpy
.
asarray
(
y
)
if
all
(
a
.
dtype
in
discrete_types
for
a
in
(
x
,
y
)):
if
all
(
a
.
dtype
in
discrete_types
for
a
in
(
x
,
y
)):
return
numpy
.
array
(
float
(
x
)
/
y
,
dtype
=
config
.
floatX
)
return
numpy
.
sctypeDict
[
config
.
floatX
](
float
(
x
)
/
y
)
else
:
else
:
return
x
/
y
return
x
/
y
...
@@ -2166,7 +2166,7 @@ neg = Neg(same_out, name='neg')
...
@@ -2166,7 +2166,7 @@ neg = Neg(same_out, name='neg')
class
Inv
(
UnaryScalarOp
):
class
Inv
(
UnaryScalarOp
):
""" multiplicative inverse. Also called reciprocal"""
""" multiplicative inverse. Also called reciprocal"""
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
return
1.0
/
x
return
numpy
.
float32
(
1.0
)
/
x
def
grad
(
self
,
(
x
,),
(
gz
,)):
def
grad
(
self
,
(
x
,),
(
gz
,)):
if
x
.
type
in
complex_types
:
if
x
.
type
in
complex_types
:
...
@@ -2180,6 +2180,8 @@ class Inv(UnaryScalarOp):
...
@@ -2180,6 +2180,8 @@ class Inv(UnaryScalarOp):
return
-
gz
/
(
x
*
x
),
return
-
gz
/
(
x
*
x
),
def
c_code
(
self
,
node
,
name
,
(
x
,),
(
z
,),
sub
):
def
c_code
(
self
,
node
,
name
,
(
x
,),
(
z
,),
sub
):
if
node
.
inputs
[
0
]
.
type
in
complex_types
:
raise
NotImplementedError
()
return
"
%(z)
s = 1.0 /
%(x)
s;"
%
locals
()
return
"
%(z)
s = 1.0 /
%(x)
s;"
%
locals
()
inv
=
Inv
(
upgrade_to_float
,
name
=
'inv'
)
inv
=
Inv
(
upgrade_to_float
,
name
=
'inv'
)
...
@@ -2190,6 +2192,11 @@ class Log(UnaryScalarOp):
...
@@ -2190,6 +2192,11 @@ class Log(UnaryScalarOp):
amd_float64
=
"amd_vrda_log"
amd_float64
=
"amd_vrda_log"
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
# If x is an int8 or uint8, numpy.log will compute the result in
# half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
return
numpy
.
log
(
x
,
sig
=
'f'
)
return
numpy
.
log
(
x
)
return
numpy
.
log
(
x
)
def
grad
(
self
,
(
x
,),
(
gz
,)):
def
grad
(
self
,
(
x
,),
(
gz
,)):
...
@@ -2219,6 +2226,11 @@ class Log2(UnaryScalarOp):
...
@@ -2219,6 +2226,11 @@ class Log2(UnaryScalarOp):
amd_float64
=
"amd_vrda_log2"
amd_float64
=
"amd_vrda_log2"
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
# If x is an int8 or uint8, numpy.log2 will compute the result in
# half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
return
numpy
.
log2
(
x
,
sig
=
'f'
)
return
numpy
.
log2
(
x
)
return
numpy
.
log2
(
x
)
def
grad
(
self
,
(
x
,),
(
gz
,)):
def
grad
(
self
,
(
x
,),
(
gz
,)):
...
@@ -2245,6 +2257,11 @@ class Log10(UnaryScalarOp):
...
@@ -2245,6 +2257,11 @@ class Log10(UnaryScalarOp):
amd_float64
=
"amd_vrda_log10"
amd_float64
=
"amd_vrda_log10"
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
# If x is an int8 or uint8, numpy.log10 will compute the result in
# half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
return
numpy
.
log10
(
x
,
sig
=
'f'
)
return
numpy
.
log10
(
x
)
return
numpy
.
log10
(
x
)
def
grad
(
self
,
(
x
,),
(
gz
,)):
def
grad
(
self
,
(
x
,),
(
gz
,)):
...
@@ -2268,6 +2285,11 @@ log10 = Log10(upgrade_to_float, name='log10')
...
@@ -2268,6 +2285,11 @@ log10 = Log10(upgrade_to_float, name='log10')
class
Log1p
(
UnaryScalarOp
):
class
Log1p
(
UnaryScalarOp
):
""" log(1+x) """
""" log(1+x) """
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
# If x is an int8 or uint8, numpy.log1p will compute the result in
# half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
return
numpy
.
log1p
(
x
,
sig
=
'f'
)
return
numpy
.
log1p
(
x
)
return
numpy
.
log1p
(
x
)
def
grad
(
self
,
(
x
,),
(
gz
,)):
def
grad
(
self
,
(
x
,),
(
gz
,)):
...
@@ -2293,6 +2315,11 @@ class Exp(UnaryScalarOp):
...
@@ -2293,6 +2315,11 @@ class Exp(UnaryScalarOp):
amd_float64
=
"amd_vrda_exp"
amd_float64
=
"amd_vrda_exp"
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
# If x is an int8 or uint8, numpy.exp will compute the result in
# half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
return
numpy
.
exp
(
x
,
sig
=
'f'
)
return
numpy
.
exp
(
x
)
return
numpy
.
exp
(
x
)
def
grad
(
self
,
(
x
,
),
(
gz
,
)):
def
grad
(
self
,
(
x
,
),
(
gz
,
)):
...
@@ -2315,6 +2342,11 @@ exp = Exp(upgrade_to_float, name='exp')
...
@@ -2315,6 +2342,11 @@ exp = Exp(upgrade_to_float, name='exp')
class
Exp2
(
UnaryScalarOp
):
class
Exp2
(
UnaryScalarOp
):
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
# If x is an int8 or uint8, numpy.exp2 will compute the result in
# half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
return
numpy
.
exp2
(
x
,
sig
=
'f'
)
return
numpy
.
exp2
(
x
)
return
numpy
.
exp2
(
x
)
def
grad
(
self
,
(
x
,
),
(
gz
,
)):
def
grad
(
self
,
(
x
,
),
(
gz
,
)):
...
@@ -2337,6 +2369,11 @@ exp2 = Exp2(upgrade_to_float, name='exp2')
...
@@ -2337,6 +2369,11 @@ exp2 = Exp2(upgrade_to_float, name='exp2')
class
Expm1
(
UnaryScalarOp
):
class
Expm1
(
UnaryScalarOp
):
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
# If x is an int8 or uint8, numpy.expm1 will compute the result in
# half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
return
numpy
.
expm1
(
x
,
sig
=
'f'
)
return
numpy
.
expm1
(
x
)
return
numpy
.
expm1
(
x
)
def
grad
(
self
,
(
x
,
),
(
gz
,
)):
def
grad
(
self
,
(
x
,
),
(
gz
,
)):
...
@@ -2382,6 +2419,11 @@ sqr = Sqr(same_out, name='sqr')
...
@@ -2382,6 +2419,11 @@ sqr = Sqr(same_out, name='sqr')
class
Sqrt
(
UnaryScalarOp
):
class
Sqrt
(
UnaryScalarOp
):
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
# If x is an int8 or uint8, numpy.sqrt will compute the result in
# half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
return
numpy
.
sqrt
(
x
,
sig
=
'f'
)
return
numpy
.
sqrt
(
x
)
return
numpy
.
sqrt
(
x
)
def
grad
(
self
,
(
x
,),
(
gz
,)):
def
grad
(
self
,
(
x
,),
(
gz
,)):
...
@@ -2404,6 +2446,11 @@ sqrt = Sqrt(upgrade_to_float, name='sqrt')
...
@@ -2404,6 +2446,11 @@ sqrt = Sqrt(upgrade_to_float, name='sqrt')
class
Deg2Rad
(
UnaryScalarOp
):
class
Deg2Rad
(
UnaryScalarOp
):
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
# If x is an int8 or uint8, numpy.deg2rad will compute the result in
# half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
return
numpy
.
deg2rad
(
x
,
sig
=
'f'
)
return
numpy
.
deg2rad
(
x
)
return
numpy
.
deg2rad
(
x
)
def
grad
(
self
,
(
x
,),
(
gz
,)):
def
grad
(
self
,
(
x
,),
(
gz
,)):
...
@@ -2426,6 +2473,11 @@ deg2rad = Deg2Rad(upgrade_to_float, name='deg2rad')
...
@@ -2426,6 +2473,11 @@ deg2rad = Deg2Rad(upgrade_to_float, name='deg2rad')
class
Rad2Deg
(
UnaryScalarOp
):
class
Rad2Deg
(
UnaryScalarOp
):
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
# If x is an int8 or uint8, numpy.rad2deg will compute the result in
# half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
return
numpy
.
rad2deg
(
x
,
sig
=
'f'
)
return
numpy
.
rad2deg
(
x
)
return
numpy
.
rad2deg
(
x
)
def
grad
(
self
,
(
x
,),
(
gz
,)):
def
grad
(
self
,
(
x
,),
(
gz
,)):
...
@@ -2451,6 +2503,11 @@ class Cos(UnaryScalarOp):
...
@@ -2451,6 +2503,11 @@ class Cos(UnaryScalarOp):
amd_float64
=
"amd_vrda_cos"
amd_float64
=
"amd_vrda_cos"
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
# If x is an int8 or uint8, numpy.cos will compute the result in
# half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
return
numpy
.
cos
(
x
,
sig
=
'f'
)
return
numpy
.
cos
(
x
)
return
numpy
.
cos
(
x
)
def
grad
(
self
,
(
x
,
),
(
gz
,
)):
def
grad
(
self
,
(
x
,
),
(
gz
,
)):
...
@@ -2473,6 +2530,11 @@ cos = Cos(upgrade_to_float, name='cos')
...
@@ -2473,6 +2530,11 @@ cos = Cos(upgrade_to_float, name='cos')
class
ArcCos
(
UnaryScalarOp
):
class
ArcCos
(
UnaryScalarOp
):
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
# If x is an int8 or uint8, numpy.arccos will compute the result in
# half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
return
numpy
.
arccos
(
x
,
sig
=
'f'
)
return
numpy
.
arccos
(
x
)
return
numpy
.
arccos
(
x
)
def
grad
(
self
,
(
x
,),
(
gz
,)):
def
grad
(
self
,
(
x
,),
(
gz
,)):
...
@@ -2498,6 +2560,11 @@ class Sin(UnaryScalarOp):
...
@@ -2498,6 +2560,11 @@ class Sin(UnaryScalarOp):
amd_float64
=
"amd_vrda_sin"
amd_float64
=
"amd_vrda_sin"
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
# If x is an int8 or uint8, numpy.sin will compute the result in
# half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
return
numpy
.
sin
(
x
,
sig
=
'f'
)
return
numpy
.
sin
(
x
)
return
numpy
.
sin
(
x
)
def
grad
(
self
,
(
x
,
),
(
gz
,
)):
def
grad
(
self
,
(
x
,
),
(
gz
,
)):
...
@@ -2520,6 +2587,11 @@ sin = Sin(upgrade_to_float, name='sin')
...
@@ -2520,6 +2587,11 @@ sin = Sin(upgrade_to_float, name='sin')
class
ArcSin
(
UnaryScalarOp
):
class
ArcSin
(
UnaryScalarOp
):
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
# If x is an int8 or uint8, numpy.arcsin will compute the result in
# half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
return
numpy
.
arcsin
(
x
,
sig
=
'f'
)
return
numpy
.
arcsin
(
x
)
return
numpy
.
arcsin
(
x
)
def
grad
(
self
,
(
x
,),
(
gz
,)):
def
grad
(
self
,
(
x
,),
(
gz
,)):
...
@@ -2542,6 +2614,11 @@ arcsin = ArcSin(upgrade_to_float, name='arcsin')
...
@@ -2542,6 +2614,11 @@ arcsin = ArcSin(upgrade_to_float, name='arcsin')
class
Tan
(
UnaryScalarOp
):
class
Tan
(
UnaryScalarOp
):
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
# If x is an int8 or uint8, numpy.tan will compute the result in
# half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
return
numpy
.
tan
(
x
,
sig
=
'f'
)
return
numpy
.
tan
(
x
)
return
numpy
.
tan
(
x
)
def
grad
(
self
,
(
x
,),
(
gz
,)):
def
grad
(
self
,
(
x
,),
(
gz
,)):
...
@@ -2564,6 +2641,11 @@ tan = Tan(upgrade_to_float, name='tan')
...
@@ -2564,6 +2641,11 @@ tan = Tan(upgrade_to_float, name='tan')
class
ArcTan
(
UnaryScalarOp
):
class
ArcTan
(
UnaryScalarOp
):
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
# If x is an int8 or uint8, numpy.arctan will compute the result in
# half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
return
numpy
.
arctan
(
x
,
sig
=
'f'
)
return
numpy
.
arctan
(
x
)
return
numpy
.
arctan
(
x
)
def
grad
(
self
,
(
x
,),
(
gz
,)):
def
grad
(
self
,
(
x
,),
(
gz
,)):
...
@@ -2586,6 +2668,13 @@ arctan = ArcTan(upgrade_to_float, name='arctan')
...
@@ -2586,6 +2668,13 @@ arctan = ArcTan(upgrade_to_float, name='arctan')
class
ArcTan2
(
BinaryScalarOp
):
class
ArcTan2
(
BinaryScalarOp
):
def
impl
(
self
,
y
,
x
):
def
impl
(
self
,
y
,
x
):
# If x and y are int8 or uint8, numpy.arctan2 will compute the result
# in half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
y_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
y_dtype
in
(
'int8'
,
'uint8'
):
return
numpy
.
arctan2
(
y
,
x
,
sig
=
'f'
)
return
numpy
.
arctan2
(
y
,
x
)
return
numpy
.
arctan2
(
y
,
x
)
def
grad
(
self
,
(
y
,
x
),
(
gz
,)):
def
grad
(
self
,
(
y
,
x
),
(
gz
,)):
...
@@ -2621,6 +2710,11 @@ class Cosh(UnaryScalarOp):
...
@@ -2621,6 +2710,11 @@ class Cosh(UnaryScalarOp):
cosh(x) = (exp(x) + exp(-x)) / 2
cosh(x) = (exp(x) + exp(-x)) / 2
"""
"""
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
# If x is an int8 or uint8, numpy.cosh will compute the result in
# half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
return
numpy
.
cosh
(
x
,
sig
=
'f'
)
return
numpy
.
cosh
(
x
)
return
numpy
.
cosh
(
x
)
def
grad
(
self
,
(
x
,
),
(
gz
,
)):
def
grad
(
self
,
(
x
,
),
(
gz
,
)):
...
@@ -2643,6 +2737,11 @@ cosh = Cosh(upgrade_to_float, name='cosh')
...
@@ -2643,6 +2737,11 @@ cosh = Cosh(upgrade_to_float, name='cosh')
class
ArcCosh
(
UnaryScalarOp
):
class
ArcCosh
(
UnaryScalarOp
):
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
# If x is an int8 or uint8, numpy.arccosh will compute the result in
# half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
return
numpy
.
arccosh
(
x
,
sig
=
'f'
)
return
numpy
.
arccosh
(
x
)
return
numpy
.
arccosh
(
x
)
def
grad
(
self
,
(
x
,
),
(
gz
,
)):
def
grad
(
self
,
(
x
,
),
(
gz
,
)):
...
@@ -2668,6 +2767,11 @@ class Sinh(UnaryScalarOp):
...
@@ -2668,6 +2767,11 @@ class Sinh(UnaryScalarOp):
sinh(x) = (exp(x) - exp(-x)) / 2
sinh(x) = (exp(x) - exp(-x)) / 2
"""
"""
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
# If x is an int8 or uint8, numpy.sinh will compute the result in
# half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
return
numpy
.
sinh
(
x
,
sig
=
'f'
)
return
numpy
.
sinh
(
x
)
return
numpy
.
sinh
(
x
)
def
grad
(
self
,
(
x
,
),
(
gz
,
)):
def
grad
(
self
,
(
x
,
),
(
gz
,
)):
...
@@ -2690,6 +2794,11 @@ sinh = Sinh(upgrade_to_float, name='sinh')
...
@@ -2690,6 +2794,11 @@ sinh = Sinh(upgrade_to_float, name='sinh')
class
ArcSinh
(
UnaryScalarOp
):
class
ArcSinh
(
UnaryScalarOp
):
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
# If x is an int8 or uint8, numpy.arcsinh will compute the result in
# half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
return
numpy
.
arcsinh
(
x
,
sig
=
'f'
)
return
numpy
.
arcsinh
(
x
)
return
numpy
.
arcsinh
(
x
)
def
grad
(
self
,
(
x
,
),
(
gz
,
)):
def
grad
(
self
,
(
x
,
),
(
gz
,
)):
...
@@ -2716,6 +2825,11 @@ class Tanh(UnaryScalarOp):
...
@@ -2716,6 +2825,11 @@ class Tanh(UnaryScalarOp):
= (exp(2*x) - 1) / (exp(2*x) + 1)
= (exp(2*x) - 1) / (exp(2*x) + 1)
"""
"""
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
# If x is an int8 or uint8, numpy.tanh will compute the result in
# half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
return
numpy
.
tanh
(
x
,
sig
=
'f'
)
return
numpy
.
tanh
(
x
)
return
numpy
.
tanh
(
x
)
def
grad
(
self
,
(
x
,
),
(
gz
,
)):
def
grad
(
self
,
(
x
,
),
(
gz
,
)):
...
@@ -2738,6 +2852,11 @@ tanh = Tanh(upgrade_to_float, name='tanh')
...
@@ -2738,6 +2852,11 @@ tanh = Tanh(upgrade_to_float, name='tanh')
class
ArcTanh
(
UnaryScalarOp
):
class
ArcTanh
(
UnaryScalarOp
):
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
# If x is an int8 or uint8, numpy.arctanh will compute the result in
# half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
return
numpy
.
arctanh
(
x
,
sig
=
'f'
)
return
numpy
.
arctanh
(
x
)
return
numpy
.
arctanh
(
x
)
def
grad
(
self
,
(
x
,
),
(
gz
,
)):
def
grad
(
self
,
(
x
,
),
(
gz
,
)):
...
...
theano/scalar/tests/test_basic.py
浏览文件 @
52cb8ec7
...
@@ -10,6 +10,7 @@ If you do want to rewrite these tests, bear in mind:
...
@@ -10,6 +10,7 @@ If you do want to rewrite these tests, bear in mind:
"""
"""
import
unittest
import
unittest
import
numpy
as
np
import
theano
import
theano
from
theano.gof
import
FunctionGraph
from
theano.gof
import
FunctionGraph
...
@@ -20,8 +21,12 @@ from theano.scalar.basic import (floats, float32, float64,
...
@@ -20,8 +21,12 @@ from theano.scalar.basic import (floats, float32, float64,
ints
,
int8
,
int32
,
complex64
,
ints
,
int8
,
int32
,
complex64
,
ComplexError
,
IntDiv
,
TrueDiv
,
ComplexError
,
IntDiv
,
TrueDiv
,
Composite
,
add
,
div_proxy
,
clip
,
Composite
,
add
,
div_proxy
,
clip
,
and_
,
eq
,
neq
,
invert
,
mul
)
and_
,
eq
,
neq
,
invert
,
mul
,
Scalar
)
import
numpy
from
theano.scalar.basic
import
(
true_div
,
inv
,
log
,
log2
,
log10
,
log1p
,
exp
,
exp2
,
expm1
,
sqrt
,
deg2rad
,
rad2deg
,
cos
,
arccos
,
sin
,
arcsin
,
tan
,
arctan
,
arctan2
,
cosh
,
arccosh
,
sinh
,
arcsinh
,
tanh
,
arctanh
)
def
inputs
():
def
inputs
():
return
floats
(
'xyz'
)
return
floats
(
'xyz'
)
...
@@ -75,7 +80,7 @@ class test_ScalarOps(unittest.TestCase):
...
@@ -75,7 +80,7 @@ class test_ScalarOps(unittest.TestCase):
g3
=
theano
.
gradient
.
grad
(
a3
,
x
)
g3
=
theano
.
gradient
.
grad
(
a3
,
x
)
fn3
=
gof
.
DualLinker
()
.
accept
(
FunctionGraph
([
x
],
[
g3
]))
.
make_function
()
fn3
=
gof
.
DualLinker
()
.
accept
(
FunctionGraph
([
x
],
[
g3
]))
.
make_function
()
rng
=
n
umpy
.
random
.
RandomState
(
utt
.
fetch_seed
())
rng
=
n
p
.
random
.
RandomState
(
utt
.
fetch_seed
())
ntests
=
50
ntests
=
50
for
i
in
xrange
(
ntests
):
for
i
in
xrange
(
ntests
):
...
@@ -235,6 +240,128 @@ class test_logical(unittest.TestCase):
...
@@ -235,6 +240,128 @@ class test_logical(unittest.TestCase):
self
.
assertTrue
(
fn
(
a
,
b
)
==
~
a
,
(
a
,))
self
.
assertTrue
(
fn
(
a
,
b
)
==
~
a
,
(
a
,))
# This class does not inherit from unittest.TestCase, because it would
# interfere with the "yield" mechanism that automatically generates test, see
# http://stackoverflow.com/questions/6689537/nose-test-generators-inside-class
# Therefore, it needs to be named "test_..." or "Test_...", so nose can pick
# it up by name, otherwise the tests would not be executed.
class
test_upgrade_to_float
(
object
):
# Test for Ops whose output has to be floating point, even when all
# inputs are ints.
# In particular, when the inputs are int8, the output should be
# at least float32, not float16.
unary_ops_vals
=
[
(
inv
,
range
(
-
127
,
0
)
+
range
(
1
,
127
)),
(
sqrt
,
range
(
0
,
128
)),
(
log
,
range
(
1
,
128
)),
(
log2
,
range
(
1
,
128
)),
(
log10
,
range
(
1
,
128
)),
(
log1p
,
range
(
0
,
128
)),
(
exp
,
range
(
-
127
,
89
)),
(
exp2
,
range
(
-
127
,
89
)),
(
expm1
,
range
(
-
127
,
89
)),
(
deg2rad
,
range
(
-
127
,
128
)),
(
rad2deg
,
range
(
-
127
,
128
)),
(
cos
,
range
(
-
127
,
128
)),
(
arccos
,
range
(
-
1
,
2
)),
(
cosh
,
range
(
-
89
,
90
)),
(
arccosh
,
range
(
1
,
128
)),
(
sin
,
range
(
-
127
,
128
)),
(
arcsin
,
range
(
-
1
,
2
)),
(
sinh
,
range
(
-
89
,
90
)),
(
arcsinh
,
range
(
-
127
,
128
)),
(
tan
,
range
(
-
3
,
4
)),
(
arctan
,
range
(
-
127
,
128
)),
(
tanh
,
range
(
-
127
,
128
)),
(
arctanh
,
[
0
])]
binary_ops_vals
=
[
(
arctan2
,
range
(
-
127
,
128
),
range
(
-
127
,
128
))]
@staticmethod
def
_test_unary
(
unary_op
,
x_range
):
xi
=
int8
(
'xi'
)
xf
=
float32
(
'xf'
)
ei
=
unary_op
(
xi
)
fi
=
theano
.
function
([
xi
],
ei
)
ef
=
unary_op
(
xf
)
ff
=
theano
.
function
([
xf
],
ef
)
for
x_val
in
x_range
:
outi
=
fi
(
x_val
)
outf
=
ff
(
x_val
)
assert
outi
.
dtype
==
outf
.
dtype
,
'incorrect dtype'
assert
np
.
allclose
(
outi
,
outf
),
'insufficient precision'
@staticmethod
def
_test_binary
(
binary_op
,
x_range
,
y_range
):
xi
=
int8
(
'xi'
)
yi
=
int8
(
'yi'
)
xf
=
float32
(
'xf'
)
yf
=
float32
(
'yf'
)
ei
=
binary_op
(
xi
,
yi
)
fi
=
theano
.
function
([
xi
,
yi
],
ei
)
ef
=
binary_op
(
xf
,
yf
)
ff
=
theano
.
function
([
xf
,
yf
],
ef
)
for
x_val
in
x_range
:
for
y_val
in
y_range
:
outi
=
fi
(
x_val
,
y_val
)
outf
=
ff
(
x_val
,
y_val
)
assert
outi
.
dtype
==
outf
.
dtype
,
'incorrect dtype'
assert
np
.
allclose
(
outi
,
outf
),
'insufficient precision'
def
test_true_div
(
self
):
# true_div's upcast policy is not exactly "upgrade_to_float",
# so the test is a little bit different
x_range
=
range
(
-
127
,
128
)
y_range
=
range
(
-
127
,
0
)
+
range
(
1
,
127
)
xi
=
int8
(
'xi'
)
yi
=
int8
(
'yi'
)
xf
=
Scalar
(
theano
.
config
.
floatX
)(
'xf'
)
yf
=
Scalar
(
theano
.
config
.
floatX
)(
'yf'
)
ei
=
true_div
(
xi
,
yi
)
fi
=
theano
.
function
([
xi
,
yi
],
ei
)
ef
=
true_div
(
xf
,
yf
)
ff
=
theano
.
function
([
xf
,
yf
],
ef
)
for
x_val
in
x_range
:
for
y_val
in
y_range
:
outi
=
fi
(
x_val
,
y_val
)
outf
=
ff
(
x_val
,
y_val
)
assert
outi
.
dtype
==
outf
.
dtype
,
'incorrect dtype'
assert
np
.
allclose
(
outi
,
outf
),
'insufficient precision'
def
test_unary
(
self
):
# Automatically define all individual unary tests
for
unary_op
,
x_range
in
self
.
unary_ops_vals
:
test_name
=
'test_
%
s'
%
unary_op
.
name
# Make a lambda function so we can name the test
test
=
lambda
:
self
.
_test_unary
(
unary_op
,
x_range
)
test
.
description
=
test_name
yield
test
def
test_binary
(
self
):
# Automatically define all individual binary tests
for
binary_op
,
x_range
,
y_range
in
self
.
binary_ops_vals
:
test_name
=
'test_
%
s'
%
binary_op
.
name
# Make a lambda function so we can name the test
test
=
lambda
:
self
.
_test_binary
(
binary_op
,
x_range
,
y_range
)
test
.
description
=
test_name
yield
test
class
test_complex_mod
(
unittest
.
TestCase
):
class
test_complex_mod
(
unittest
.
TestCase
):
"""Make sure
%
fails on complex numbers."""
"""Make sure
%
fails on complex numbers."""
...
...
theano/tensor/basic.py
浏览文件 @
52cb8ec7
...
@@ -1812,7 +1812,7 @@ def round(a, mode="half_away_from_zero"):
...
@@ -1812,7 +1812,7 @@ def round(a, mode="half_away_from_zero"):
raise
Exception
(
"round mode
%
s is not implemented."
%
mode
)
raise
Exception
(
"round mode
%
s is not implemented."
%
mode
)
@_scal_elemwise_with_nfunc
(
'around'
,
1
,
-
1
)
@_scal_elemwise_with_nfunc
(
'around'
,
1
,
1
)
def
round_half_to_even
(
a
):
def
round_half_to_even
(
a
):
"""round_half_to_even(a)"""
"""round_half_to_even(a)"""
...
@@ -1952,20 +1952,20 @@ def chi2sf(x, k):
...
@@ -1952,20 +1952,20 @@ def chi2sf(x, k):
#numpy.real(float32) return a view on the inputs.
#numpy.real(float32) return a view on the inputs.
#@_scal_elemwise_with_nfunc('real', 1,
-
1)
#@_scal_elemwise_with_nfunc('real', 1, 1)
@_scal_elemwise
@_scal_elemwise
def
real
(
z
):
def
real
(
z
):
"""Return real component of complex-valued tensor `z`"""
"""Return real component of complex-valued tensor `z`"""
_tensor_py_operators
.
real
=
property
(
real
)
_tensor_py_operators
.
real
=
property
(
real
)
@_scal_elemwise_with_nfunc
(
'imag'
,
1
,
-
1
)
@_scal_elemwise_with_nfunc
(
'imag'
,
1
,
1
)
def
imag
(
z
):
def
imag
(
z
):
"""Return imaginary component of complex-valued tensor `z`"""
"""Return imaginary component of complex-valued tensor `z`"""
_tensor_py_operators
.
imag
=
property
(
imag
)
_tensor_py_operators
.
imag
=
property
(
imag
)
@_scal_elemwise_with_nfunc
(
'angle'
,
1
,
-
1
)
@_scal_elemwise_with_nfunc
(
'angle'
,
1
,
1
)
def
angle
(
z
):
def
angle
(
z
):
"""Return polar-coordinate angle of complex-valued tensor `z`"""
"""Return polar-coordinate angle of complex-valued tensor `z`"""
...
@@ -1975,7 +1975,7 @@ def complex(real, imag):
...
@@ -1975,7 +1975,7 @@ def complex(real, imag):
"""Return complex-valued tensor with `real` and `imag` components"""
"""Return complex-valued tensor with `real` and `imag` components"""
@_scal_elemwise_with_nfunc
(
'conj'
,
1
,
-
1
)
@_scal_elemwise_with_nfunc
(
'conj'
,
1
,
1
)
def
conj
(
z
):
def
conj
(
z
):
"""Return the complex conjugate of `z`."""
"""Return the complex conjugate of `z`."""
...
...
theano/tensor/elemwise.py
浏览文件 @
52cb8ec7
...
@@ -18,9 +18,10 @@ from theano.tensor import elemwise_cgen as cgen
...
@@ -18,9 +18,10 @@ from theano.tensor import elemwise_cgen as cgen
config
=
theano
.
config
config
=
theano
.
config
# We cannot import discrete_dtypes from tensor.basic yet,
# We cannot import discrete_dtypes
or float_dtypes
from tensor.basic yet,
# so we redefine them here
# so we redefine them here
discrete_dtypes
=
map
(
str
,
scalar
.
discrete_types
)
discrete_dtypes
=
map
(
str
,
scalar
.
discrete_types
)
float_dtypes
=
map
(
str
,
scalar
.
float_types
)
# tensor depends on elemwise to provide definitions for several ops
# tensor depends on elemwise to provide definitions for several ops
...
@@ -472,14 +473,11 @@ class Elemwise(OpenMPOp):
...
@@ -472,14 +473,11 @@ class Elemwise(OpenMPOp):
the input's storage. (Just like destroymap, but without the lists.)
the input's storage. (Just like destroymap, but without the lists.)
* nfunc_spec: either None or a tuple of three elements,
* nfunc_spec: either None or a tuple of three elements,
(nfunc_name, nin, nout) such that getattr(numpy, nfunc_name)
(nfunc_name, nin, nout) such that getattr(numpy, nfunc_name)
implements this operation, takes nin inputs and abs(nout) outputs
implements this operation, takes nin inputs and nout outputs.
(nout < 0 if the numpy function does not provide the option of
Note that nin cannot always be inferred from the scalar op's
providing a numpy array to store the results in). Note that nin
own nin field because that value is sometimes 0 (meaning a
cannot always be inferred from the scalar op's own nin field
variable number of inputs), whereas the numpy function may
because that value is sometimes 0 (meaning a variable number of
not have varargs.
inputs), whereas the numpy function may not have varargs.
NOTE: as of now, the sign of the nout field is ignored (some work
needs to be done to resize the destinations when needed).
"""
"""
if
inplace_pattern
is
None
:
if
inplace_pattern
is
None
:
inplace_pattern
=
{}
inplace_pattern
=
{}
...
@@ -819,43 +817,24 @@ class Elemwise(OpenMPOp):
...
@@ -819,43 +817,24 @@ class Elemwise(OpenMPOp):
out_shape
.
append
(
max
(
values
))
out_shape
.
append
(
max
(
values
))
out_shape
=
tuple
(
out_shape
)
out_shape
=
tuple
(
out_shape
)
# Commented as we don't reuse outputs now.
ufunc_args
=
inputs
#
ufunc_kwargs
=
{}
# if not self.inplace_pattern:
# for output, storage in izip(node.outputs, output_storage):
# odat = storage[0]
# if odat is not None:
# if odat.shape != out_shape:
# # It is unsafe to try to resize odat,
# # we have to allocate output storage.
# odat = None
# if odat is None:
# odat = numpy.ndarray(out_shape, dtype=output.type.dtype)
# storage[0] = odat
# else:
# for i, (output, storage) in enumerate(
# izip(node.outputs, output_storage)):
# #i is an output idx
# if i in self.inplace_pattern:
# odat = inputs[self.inplace_pattern[i]]
# else:
# odat = storage[0]
# if odat is not None:
# if odat.shape != out_shape:
# # It is unsafe to try to resize odat,
# # we have to allocate output storage.
# odat = None
# if odat is None:
# odat = numpy.ndarray(out_shape,
# dtype=output.type.dtype)
# storage[0] = odat
ufunc_args
=
inputs
# + output_storage
if
self
.
nfunc
and
len
(
inputs
)
==
self
.
nfunc_spec
[
1
]:
if
self
.
nfunc
and
len
(
inputs
)
==
self
.
nfunc_spec
[
1
]:
ufunc
=
self
.
nfunc
ufunc
=
self
.
nfunc
nout
=
self
.
nfunc_spec
[
2
]
nout
=
self
.
nfunc_spec
[
2
]
if
nout
<
0
:
# Numpy ufuncs will sometimes perform operations in
nout
=
-
nout
# float16, in particular when the input is int8.
# This is not something that we want, and we do not
# do it in the C code, so we specify that the computation
# should be carried out in the returned dtype.
# This is done via the "sig" kwarg of the ufunc, its value
# should be something like "ff->f", where the characters
# represent the dtype of the inputs and outputs.
out_dtype
=
node
.
outputs
[
0
]
.
dtype
if
out_dtype
in
float_dtypes
and
isinstance
(
ufunc
,
numpy
.
ufunc
):
char
=
numpy
.
sctype2char
(
out_dtype
)
sig
=
char
*
node
.
nin
+
'->'
+
char
*
node
.
nout
ufunc_kwargs
[
'sig'
]
=
sig
# Unfortunately, the else case does not allow us to
# Unfortunately, the else case does not allow us to
# directly feed the destination arguments to the nfunc
# directly feed the destination arguments to the nfunc
# since it sometimes requires resizing. Doing this
# since it sometimes requires resizing. Doing this
...
@@ -869,7 +848,7 @@ class Elemwise(OpenMPOp):
...
@@ -869,7 +848,7 @@ class Elemwise(OpenMPOp):
self
.
scalar_op
.
nout
))
self
.
scalar_op
.
nout
))
nout
=
ufunc
.
nout
nout
=
ufunc
.
nout
variables
=
ufunc
(
*
ufunc_args
)
variables
=
ufunc
(
*
ufunc_args
,
**
ufunc_kwargs
)
if
nout
==
1
:
if
nout
==
1
:
variables
=
[
variables
]
variables
=
[
variables
]
...
...
theano/tensor/nnet/sigm.py
浏览文件 @
52cb8ec7
...
@@ -31,6 +31,11 @@ class ScalarSigmoid(scalar.UnaryScalarOp):
...
@@ -31,6 +31,11 @@ class ScalarSigmoid(scalar.UnaryScalarOp):
return
0.0
return
0.0
if
x
>
30.0
:
if
x
>
30.0
:
return
1.0
return
1.0
# If x is an int8 or uint8, numpy.exp will compute the result in
# half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
return
1.0
/
(
1.0
+
numpy
.
exp
(
-
x
,
sig
=
'f'
))
return
1.0
/
(
1.0
+
numpy
.
exp
(
-
x
))
return
1.0
/
(
1.0
+
numpy
.
exp
(
-
x
))
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
...
@@ -268,8 +273,11 @@ def hard_sigmoid(x):
...
@@ -268,8 +273,11 @@ def hard_sigmoid(x):
Removing the slope and shift does not make it faster.
Removing the slope and shift does not make it faster.
"""
"""
slope
=
0.2
# Use the same dtype as determined by "upgrade_to_float",
shift
=
0.5
# and perform computation in that dtype.
out_dtype
=
scalar
.
upgrade_to_float
(
scalar
.
Scalar
(
dtype
=
x
.
dtype
))[
0
]
.
dtype
slope
=
tensor
.
constant
(
0.2
,
dtype
=
out_dtype
)
shift
=
tensor
.
constant
(
0.5
,
dtype
=
out_dtype
)
x
=
(
x
*
slope
)
+
shift
x
=
(
x
*
slope
)
+
shift
x
=
tensor
.
clip
(
x
,
0
,
1
)
x
=
tensor
.
clip
(
x
,
0
,
1
)
return
x
return
x
...
@@ -300,6 +308,11 @@ class ScalarSoftplus(scalar.UnaryScalarOp):
...
@@ -300,6 +308,11 @@ class ScalarSoftplus(scalar.UnaryScalarOp):
return
0.0
return
0.0
if
x
>
30.0
:
if
x
>
30.0
:
return
x
return
x
# If x is an int8 or uint8, numpy.exp will compute the result in
# half-precision (float16), where we want float32.
x_dtype
=
str
(
getattr
(
x
,
'dtype'
,
''
))
if
x_dtype
in
(
'int8'
,
'uint8'
):
return
numpy
.
log1p
(
numpy
.
exp
(
x
,
sig
=
'f'
))
return
numpy
.
log1p
(
numpy
.
exp
(
x
))
return
numpy
.
log1p
(
numpy
.
exp
(
x
))
def
impl
(
self
,
x
):
def
impl
(
self
,
x
):
...
...
theano/tensor/nnet/tests/test_sigm.py
浏览文件 @
52cb8ec7
...
@@ -16,7 +16,7 @@ from theano.tensor.nnet.sigm import (
...
@@ -16,7 +16,7 @@ from theano.tensor.nnet.sigm import (
register_local_1msigmoid
,
simplify_mul
,
register_local_1msigmoid
,
simplify_mul
,
)
)
from
theano.tensor.tests.test_basic
import
(
makeBroadcastTester
,
rand
,
from
theano.tensor.tests.test_basic
import
(
makeBroadcastTester
,
rand
,
check_floatX
,
check_floatX
,
upcast_int8_nfunc
,
_good_broadcast_unary_normal_no_complex
)
_good_broadcast_unary_normal_no_complex
)
...
@@ -30,8 +30,8 @@ class T_sigmoid(unittest.TestCase):
...
@@ -30,8 +30,8 @@ class T_sigmoid(unittest.TestCase):
SigmoidTester
=
makeBroadcastTester
(
SigmoidTester
=
makeBroadcastTester
(
op
=
sigmoid
,
op
=
sigmoid
,
expected
=
lambda
inputs
:
check_floatX
(
expected
=
upcast_int8_nfunc
(
lambda
inputs
:
check_floatX
(
inputs
,
1
/
(
1
+
numpy
.
exp
(
-
inputs
))),
inputs
,
1
/
(
1
+
numpy
.
exp
(
-
inputs
)
))),
good
=
_good_broadcast_unary_normal_no_complex
,
good
=
_good_broadcast_unary_normal_no_complex
,
#grad=_grad_broadcast_unary_normal,
#grad=_grad_broadcast_unary_normal,
name
=
'SigmoidTester'
,
name
=
'SigmoidTester'
,
...
@@ -39,8 +39,8 @@ SigmoidTester = makeBroadcastTester(
...
@@ -39,8 +39,8 @@ SigmoidTester = makeBroadcastTester(
UltraFastSigmoidTester
=
makeBroadcastTester
(
UltraFastSigmoidTester
=
makeBroadcastTester
(
op
=
ultra_fast_sigmoid
,
op
=
ultra_fast_sigmoid
,
expected
=
lambda
inputs
:
check_floatX
(
expected
=
upcast_int8_nfunc
(
lambda
inputs
:
check_floatX
(
inputs
,
1
/
(
1
+
numpy
.
exp
(
-
inputs
))),
inputs
,
1
/
(
1
+
numpy
.
exp
(
-
inputs
)
))),
good
=
_good_broadcast_unary_normal_no_complex
,
good
=
_good_broadcast_unary_normal_no_complex
,
#grad=_grad_broadcast_unary_normal,
#grad=_grad_broadcast_unary_normal,
name
=
'UltraFastSigmoidTester'
,
name
=
'UltraFastSigmoidTester'
,
...
@@ -49,20 +49,21 @@ UltraFastSigmoidTester = makeBroadcastTester(
...
@@ -49,20 +49,21 @@ UltraFastSigmoidTester = makeBroadcastTester(
HardSigmoidTester
=
makeBroadcastTester
(
HardSigmoidTester
=
makeBroadcastTester
(
op
=
hard_sigmoid
,
op
=
hard_sigmoid
,
expected
=
lambda
inputs
:
check_floatX
(
expected
=
upcast_int8_nfunc
(
lambda
inputs
:
check_floatX
(
inputs
,
1
/
(
1
+
numpy
.
exp
(
-
inputs
))),
inputs
,
1
/
(
1
+
numpy
.
exp
(
-
inputs
)
))),
good
=
_good_broadcast_unary_normal_no_complex
,
good
=
_good_broadcast_unary_normal_no_complex
,
#grad=_grad_broadcast_unary_normal,
#grad=_grad_broadcast_unary_normal,
name
=
'
UltraFast
SigmoidTester'
,
name
=
'
Hard
SigmoidTester'
,
# This is an approx of the sigmoid. That is why we raise eps
# This is an approx of the sigmoid. That is why we raise eps
eps
=
1e-1
)
eps
=
1e-1
)
SoftplusTester
=
makeBroadcastTester
(
SoftplusTester
=
makeBroadcastTester
(
op
=
softplus
,
op
=
softplus
,
expected
=
lambda
inputs
:
check_floatX
(
expected
=
upcast_int8_nfunc
(
lambda
inputs
:
check_floatX
(
inputs
,
numpy
.
log1p
(
numpy
.
exp
(
inputs
))),
inputs
,
numpy
.
log1p
(
numpy
.
exp
(
inputs
)))),
good
=
_good_broadcast_unary_normal_no_complex
,
good
=
dict
(
_good_broadcast_unary_normal_no_complex
,
int8
=
[
numpy
.
arange
(
-
127
,
89
,
dtype
=
'int8'
)]),
#grad=_grad_broadcast_unary_normal,
#grad=_grad_broadcast_unary_normal,
name
=
'SoftplusTester'
,
name
=
'SoftplusTester'
,
)
)
...
...
theano/tensor/tests/test_basic.py
浏览文件 @
52cb8ec7
...
@@ -189,6 +189,50 @@ def safe_make_node(op, *inputs):
...
@@ -189,6 +189,50 @@ def safe_make_node(op, *inputs):
return
node
.
owner
return
node
.
owner
def
upcast_float16_ufunc
(
fn
):
"""Decorator that enforces computation is not done in float16 by NumPy.
Some ufuncs in NumPy will compute float values on int8 and uint8
in half-precision (float16), which is not enough, and not compatible
with the C code.
:param fn: numpy ufunc
:returns: function similar to fn.__call__, computing the same
value with a minimum floating-point precision of float32
"""
def
ret
(
*
args
,
**
kwargs
):
out_dtype
=
numpy
.
find_common_type
(
[
a
.
dtype
for
a
in
args
],
[
numpy
.
float16
])
if
out_dtype
==
'float16'
:
# Force everything to float32
sig
=
'f'
*
fn
.
nin
+
'->'
+
'f'
*
fn
.
nout
kwargs
.
update
(
sig
=
sig
)
return
fn
(
*
args
,
**
kwargs
)
return
ret
def
upcast_int8_nfunc
(
fn
):
"""Decorator that upcasts input of dtype int8 to float32.
This is so that floating-point computation is not carried using
half-precision (float16), as some NumPy functions do.
:param fn: function computing a floating-point value from inputs
:returns: function similar to fn, but upcasting its uint8 and int8
inputs before carrying out the computation.
"""
def
ret
(
*
args
,
**
kwargs
):
args
=
list
(
args
)
for
i
,
a
in
enumerate
(
args
):
if
getattr
(
a
,
'dtype'
,
None
)
in
(
'int8'
,
'uint8'
):
args
[
i
]
=
a
.
astype
(
'float32'
)
return
fn
(
*
args
,
**
kwargs
)
return
ret
def
makeTester
(
name
,
op
,
expected
,
checks
=
None
,
good
=
None
,
bad_build
=
None
,
def
makeTester
(
name
,
op
,
expected
,
checks
=
None
,
good
=
None
,
bad_build
=
None
,
bad_runtime
=
None
,
grad
=
None
,
mode
=
None
,
grad_rtol
=
None
,
bad_runtime
=
None
,
grad
=
None
,
mode
=
None
,
grad_rtol
=
None
,
eps
=
1e-10
,
skip
=
False
,
test_memmap
=
True
,
check_name
=
True
):
eps
=
1e-10
,
skip
=
False
,
test_memmap
=
True
,
check_name
=
True
):
...
@@ -321,7 +365,8 @@ def makeTester(name, op, expected, checks=None, good=None, bad_build=None,
...
@@ -321,7 +365,8 @@ def makeTester(name, op, expected, checks=None, good=None, bad_build=None,
expecteds
=
self
.
expected
(
*
inputs
)
expecteds
=
self
.
expected
(
*
inputs
)
eps
=
1e-10
eps
=
1e-10
if
any
([
i
.
dtype
==
'float32'
for
i
in
inputs
]):
if
any
([
i
.
dtype
in
(
'float32'
,
'int8'
,
'uint8'
)
for
i
in
inputs
]):
eps
=
1e-6
eps
=
1e-6
eps
=
numpy
.
max
([
eps
,
_eps
])
eps
=
numpy
.
max
([
eps
,
_eps
])
...
@@ -788,6 +833,9 @@ _good_broadcast_div_mod_normal_float_no_complex = dict(
...
@@ -788,6 +833,9 @@ _good_broadcast_div_mod_normal_float_no_complex = dict(
integer
=
(
randint
(
2
,
3
),
randint_nonzero
(
2
,
3
)),
integer
=
(
randint
(
2
,
3
),
randint_nonzero
(
2
,
3
)),
uinteger
=
(
randint
(
2
,
3
)
.
astype
(
"uint8"
),
uinteger
=
(
randint
(
2
,
3
)
.
astype
(
"uint8"
),
randint_nonzero
(
2
,
3
)
.
astype
(
"uint8"
)),
randint_nonzero
(
2
,
3
)
.
astype
(
"uint8"
)),
int8
=
[
numpy
.
tile
(
numpy
.
arange
(
-
127
,
128
,
dtype
=
'int8'
),
[
254
,
1
])
.
T
,
numpy
.
tile
(
numpy
.
array
(
range
(
-
127
,
0
)
+
range
(
1
,
128
),
dtype
=
'int8'
),
[
255
,
1
])],
# This empty2 doesn't work for some tests. I don't remember why
# This empty2 doesn't work for some tests. I don't remember why
#empty2=(numpy.asarray([0]), numpy.asarray([])),
#empty2=(numpy.asarray([0]), numpy.asarray([])),
)
)
...
@@ -853,7 +901,7 @@ def _numpy_true_div(x, y):
...
@@ -853,7 +901,7 @@ def _numpy_true_div(x, y):
TrueDivTester
=
makeBroadcastTester
(
TrueDivTester
=
makeBroadcastTester
(
op
=
tensor
.
true_div
,
op
=
tensor
.
true_div
,
expected
=
_numpy_true_div
,
expected
=
_numpy_true_div
,
good
=
_good_broadcast_div_mod_normal_float
,
good
=
_good_broadcast_div_mod_normal_float
_no_complex
,
grad
=
_grad_broadcast_div_mod_normal
,
grad
=
_grad_broadcast_div_mod_normal
,
grad_rtol
=
div_grad_rtol
,
grad_rtol
=
div_grad_rtol
,
)
)
...
@@ -864,12 +912,48 @@ TrueDivInplaceTester = makeBroadcastTester(
...
@@ -864,12 +912,48 @@ TrueDivInplaceTester = makeBroadcastTester(
good
=
copymod
(
good
=
copymod
(
_good_broadcast_div_mod_normal_float_inplace
,
_good_broadcast_div_mod_normal_float_inplace
,
# The output is now in float, we cannot work inplace on an int.
# The output is now in float, we cannot work inplace on an int.
without
=
[
'integer'
,
'uinteger'
]),
without
=
[
'integer'
,
'uinteger'
,
'int8'
]),
grad
=
_grad_broadcast_div_mod_normal
,
grad
=
_grad_broadcast_div_mod_normal
,
grad_rtol
=
div_grad_rtol
,
grad_rtol
=
div_grad_rtol
,
inplace
=
True
)
inplace
=
True
)
_good_inv
=
dict
(
normal
=
[
5
*
rand_nonzero
((
2
,
3
))],
integers
=
[
randint_nonzero
(
2
,
3
)],
int8
=
[
numpy
.
array
(
range
(
-
127
,
0
)
+
range
(
1
,
127
),
dtype
=
'int8'
)],
complex
=
[
randcomplex_nonzero
((
2
,
3
))],
empty
=
[
numpy
.
asarray
([],
dtype
=
config
.
floatX
)])
_good_inv_inplace
=
copymod
(
_good_inv
,
without
=
[
'integers'
,
'int8'
,
'complex'
])
_grad_inv
=
copymod
(
_good_inv
,
without
=
[
'integers'
,
'int8'
,
'complex'
,
'empty'
])
_bad_runtime_inv
=
dict
(
float
=
[
numpy
.
zeros
((
2
,
3
))],
integers
=
[
numpy
.
zeros
((
2
,
3
),
dtype
=
'int64'
)],
int8
=
[
numpy
.
zeros
((
2
,
3
),
dtype
=
'int8'
)],
complex
=
[
numpy
.
zeros
((
2
,
3
),
dtype
=
'complex128'
)])
InvTester
=
makeBroadcastTester
(
op
=
tensor
.
inv
,
expected
=
lambda
x
:
upcast_int8_nfunc
(
numpy
.
true_divide
)(
numpy
.
int8
(
1
),
x
),
good
=
_good_inv
,
bad_runtime
=
_bad_runtime_inv
,
grad
=
_grad_inv
,
grad_rtol
=
div_grad_rtol
)
InvInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
inv_inplace
,
expected
=
lambda
x
:
_numpy_true_div
(
numpy
.
int8
(
1
),
x
),
good
=
_good_inv_inplace
,
bad_runtime
=
_bad_runtime_inv
,
grad
=
_grad_inv
,
grad_rtol
=
div_grad_rtol
,
inplace
=
True
)
CeilIntDivTester
=
makeBroadcastTester
(
CeilIntDivTester
=
makeBroadcastTester
(
op
=
tensor
.
ceil_intdiv
,
op
=
tensor
.
ceil_intdiv
,
expected
=
lambda
x
,
y
:
check_floatX
((
x
,
y
),
(
x
//
y
)
+
((
x
%
y
)
!=
0
)),
expected
=
lambda
x
,
y
:
check_floatX
((
x
,
y
),
(
x
//
y
)
+
((
x
%
y
)
!=
0
)),
...
@@ -990,6 +1074,8 @@ _good_broadcast_unary_normal = dict(
...
@@ -990,6 +1074,8 @@ _good_broadcast_unary_normal = dict(
normal
=
[
numpy
.
asarray
(
rand_ranged
(
-
5
,
5
,
(
2
,
3
)),
normal
=
[
numpy
.
asarray
(
rand_ranged
(
-
5
,
5
,
(
2
,
3
)),
dtype
=
config
.
floatX
)],
dtype
=
config
.
floatX
)],
integers
=
[
randint_ranged
(
-
5
,
5
,
(
2
,
3
))],
integers
=
[
randint_ranged
(
-
5
,
5
,
(
2
,
3
))],
# not using -128 because numpy.allclose would return False
int8
=
[
numpy
.
arange
(
-
127
,
128
,
dtype
=
'int8'
)],
corner_case
=
[
corner_case
],
corner_case
=
[
corner_case
],
complex
=
[
randcomplex
(
2
,
3
)],
complex
=
[
randcomplex
(
2
,
3
)],
empty
=
[
numpy
.
asarray
([],
dtype
=
config
.
floatX
)],
empty
=
[
numpy
.
asarray
([],
dtype
=
config
.
floatX
)],
...
@@ -998,6 +1084,7 @@ _good_broadcast_unary_normal = dict(
...
@@ -998,6 +1084,7 @@ _good_broadcast_unary_normal = dict(
_good_broadcast_unary_normal_no_complex
=
dict
(
_good_broadcast_unary_normal_no_complex
=
dict
(
normal
=
[
numpy
.
asarray
(
rand_ranged
(
-
5
,
5
,
(
2
,
3
)),
dtype
=
floatX
)],
normal
=
[
numpy
.
asarray
(
rand_ranged
(
-
5
,
5
,
(
2
,
3
)),
dtype
=
floatX
)],
integers
=
[
randint_ranged
(
-
5
,
5
,
(
2
,
3
))],
integers
=
[
randint_ranged
(
-
5
,
5
,
(
2
,
3
))],
int8
=
[
numpy
.
arange
(
-
127
,
128
,
dtype
=
'int8'
)],
corner_case
=
[
corner_case
],
corner_case
=
[
corner_case
],
empty
=
[
numpy
.
asarray
([],
dtype
=
config
.
floatX
)],
empty
=
[
numpy
.
asarray
([],
dtype
=
config
.
floatX
)],
)
)
...
@@ -1020,6 +1107,8 @@ _grad_broadcast_unary_0_2_no_complex = dict(
...
@@ -1020,6 +1107,8 @@ _grad_broadcast_unary_0_2_no_complex = dict(
normal
=
[
numpy
.
asarray
(
rand_ranged
(
0
,
2
,
(
2
,
3
)),
dtype
=
floatX
)],
normal
=
[
numpy
.
asarray
(
rand_ranged
(
0
,
2
,
(
2
,
3
)),
dtype
=
floatX
)],
)
)
#inplace ops when the input is integer and the output is float*
# don't have a well defined behavior. We don't test that case.
AbsTester
=
makeBroadcastTester
(
op
=
tensor
.
abs_
,
AbsTester
=
makeBroadcastTester
(
op
=
tensor
.
abs_
,
expected
=
lambda
x
:
abs
(
x
),
expected
=
lambda
x
:
abs
(
x
),
...
@@ -1160,112 +1249,123 @@ SqrInplaceTester = makeBroadcastTester(op=inplace.sqr_inplace,
...
@@ -1160,112 +1249,123 @@ SqrInplaceTester = makeBroadcastTester(op=inplace.sqr_inplace,
grad
=
_grad_broadcast_unary_normal
,
grad
=
_grad_broadcast_unary_normal
,
inplace
=
True
)
inplace
=
True
)
ExpTester
=
makeBroadcastTester
(
op
=
tensor
.
exp
,
ExpTester
=
makeBroadcastTester
(
expected
=
numpy
.
exp
,
op
=
tensor
.
exp
,
good
=
_good_broadcast_unary_normal
,
expected
=
upcast_float16_ufunc
(
numpy
.
exp
),
good
=
dict
(
_good_broadcast_unary_normal
,
int8
=
[
numpy
.
arange
(
-
127
,
89
,
dtype
=
'int8'
)]),
grad
=
_grad_broadcast_unary_normal
)
grad
=
_grad_broadcast_unary_normal
)
ExpInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
exp_inplace
,
ExpInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
exp_inplace
,
expected
=
numpy
.
exp
,
expected
=
numpy
.
exp
,
good
=
_good_broadcast_unary_normal
,
good
=
_good_broadcast_unary_normal_float
,
grad
=
_grad_broadcast_unary_normal
,
grad
=
_grad_broadcast_unary_normal
,
inplace
=
True
)
inplace
=
True
)
def
_numpy_exp2_round_int
(
x
):
# Make sure exp2 on an int returns a value that can be correctly casted
# to an int. For instance, numpy.exp2(4) sometimes returns
# 15.999999999999998, we make sure we return 16. instead.
# This is used in Exp2InplaceTester.
out
=
numpy
.
exp2
(
x
)
if
x
.
dtype
in
tensor
.
discrete_dtypes
:
out
=
numpy
.
round
(
out
)
return
out
Exp2Tester
=
makeBroadcastTester
(
op
=
tensor
.
exp2
,
Exp2Tester
=
makeBroadcastTester
(
op
=
tensor
.
exp2
,
expected
=
numpy
.
exp2
,
expected
=
upcast_float16_ufunc
(
numpy
.
exp2
)
,
good
=
_good_broadcast_unary_normal
,
good
=
_good_broadcast_unary_normal
,
grad
=
_grad_broadcast_unary_normal
)
grad
=
_grad_broadcast_unary_normal
)
Exp2InplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
exp2_inplace
,
Exp2InplaceTester
=
makeBroadcastTester
(
expected
=
_numpy_exp2_round_int
,
op
=
inplace
.
exp2_inplace
,
good
=
_good_broadcast_unary_normal
,
expected
=
numpy
.
exp2
,
good
=
_good_broadcast_unary_normal_float
,
grad
=
_grad_broadcast_unary_normal
,
grad
=
_grad_broadcast_unary_normal
,
inplace
=
True
)
inplace
=
True
)
Expm1Tester
=
makeBroadcastTester
(
op
=
tensor
.
expm1
,
Expm1Tester
=
makeBroadcastTester
(
expected
=
numpy
.
expm1
,
op
=
tensor
.
expm1
,
good
=
_good_broadcast_unary_normal
,
expected
=
upcast_float16_ufunc
(
numpy
.
expm1
),
good
=
dict
(
_good_broadcast_unary_normal
,
int8
=
[
numpy
.
arange
(
-
127
,
89
,
dtype
=
'int8'
)]),
grad
=
_grad_broadcast_unary_normal
)
grad
=
_grad_broadcast_unary_normal
)
Expm1InplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
expm1_inplace
,
Expm1InplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
expm1_inplace
,
expected
=
numpy
.
expm1
,
expected
=
numpy
.
expm1
,
good
=
_good_broadcast_unary_normal
,
good
=
_good_broadcast_unary_normal_float
,
grad
=
_grad_broadcast_unary_normal
,
grad
=
_grad_broadcast_unary_normal
,
inplace
=
True
)
inplace
=
True
)
_good_broadcast_unary_positive
=
dict
(
normal
=
(
rand_ranged
(
0.001
,
5
,
(
2
,
3
)),),
_good_broadcast_unary_positive
=
dict
(
normal
=
(
rand_ranged
(
0.001
,
5
,
(
2
,
3
)),),
integers
=
(
randint_ranged
(
1
,
5
,
(
2
,
3
)),),
integers
=
(
randint_ranged
(
1
,
5
,
(
2
,
3
)),),
uint8
=
[
numpy
.
arange
(
1
,
256
,
dtype
=
'uint8'
)],
complex
=
(
randc128_ranged
(
1
,
5
,
(
2
,
3
)),),
complex
=
(
randc128_ranged
(
1
,
5
,
(
2
,
3
)),),
empty
=
(
numpy
.
asarray
([],
dtype
=
config
.
floatX
),),
empty
=
(
numpy
.
asarray
([],
dtype
=
config
.
floatX
),),
)
)
_good_broadcast_unary_positive_float
=
copymod
(
_good_broadcast_unary_positive
,
without
=
[
'integers'
,
'uint8'
])
_grad_broadcast_unary_positive
=
dict
(
normal
=
(
rand_ranged
(
0.001
,
5
,
(
2
,
3
)),),)
_grad_broadcast_unary_positive
=
dict
(
normal
=
(
rand_ranged
(
0.001
,
5
,
(
2
,
3
)),),)
LogTester
=
makeBroadcastTester
(
op
=
tensor
.
log
,
LogTester
=
makeBroadcastTester
(
op
=
tensor
.
log
,
expected
=
numpy
.
log
,
expected
=
upcast_float16_ufunc
(
numpy
.
log
)
,
good
=
_good_broadcast_unary_positive
,
good
=
_good_broadcast_unary_positive
,
grad
=
_grad_broadcast_unary_positive
)
grad
=
_grad_broadcast_unary_positive
)
LogInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
log_inplace
,
LogInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
log_inplace
,
expected
=
numpy
.
log
,
expected
=
numpy
.
log
,
good
=
_good_broadcast_unary_positive
,
good
=
_good_broadcast_unary_positive_float
,
grad
=
_grad_broadcast_unary_positive
,
grad
=
_grad_broadcast_unary_positive
,
inplace
=
True
)
inplace
=
True
)
Log2Tester
=
makeBroadcastTester
(
op
=
tensor
.
log2
,
Log2Tester
=
makeBroadcastTester
(
op
=
tensor
.
log2
,
expected
=
numpy
.
log2
,
expected
=
upcast_float16_ufunc
(
numpy
.
log2
)
,
good
=
_good_broadcast_unary_positive
,
good
=
_good_broadcast_unary_positive
,
grad
=
_grad_broadcast_unary_positive
)
grad
=
_grad_broadcast_unary_positive
)
Log2InplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
log2_inplace
,
Log2InplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
log2_inplace
,
expected
=
numpy
.
log2
,
expected
=
numpy
.
log2
,
good
=
_good_broadcast_unary_positive
,
good
=
_good_broadcast_unary_positive_float
,
grad
=
_grad_broadcast_unary_positive
,
grad
=
_grad_broadcast_unary_positive
,
inplace
=
True
)
inplace
=
True
)
Log10Tester
=
makeBroadcastTester
(
op
=
tensor
.
log10
,
Log10Tester
=
makeBroadcastTester
(
op
=
tensor
.
log10
,
expected
=
numpy
.
log10
,
expected
=
upcast_float16_ufunc
(
numpy
.
log10
)
,
good
=
_good_broadcast_unary_positive
,
good
=
_good_broadcast_unary_positive
,
grad
=
_grad_broadcast_unary_positive
)
grad
=
_grad_broadcast_unary_positive
)
Log10InplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
log10_inplace
,
Log10InplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
log10_inplace
,
expected
=
numpy
.
log10
,
expected
=
numpy
.
log10
,
good
=
_good_broadcast_unary_positive
,
good
=
_good_broadcast_unary_positive_float
,
grad
=
_grad_broadcast_unary_positive
,
grad
=
_grad_broadcast_unary_positive
,
inplace
=
True
)
inplace
=
True
)
Log1pTester
=
makeBroadcastTester
(
op
=
tensor
.
log1p
,
Log1pTester
=
makeBroadcastTester
(
op
=
tensor
.
log1p
,
expected
=
numpy
.
log1p
,
expected
=
upcast_float16_ufunc
(
numpy
.
log1p
)
,
good
=
_good_broadcast_unary_positive
,
good
=
_good_broadcast_unary_positive
,
grad
=
_grad_broadcast_unary_positive
)
grad
=
_grad_broadcast_unary_positive
)
Log1pInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
log1p_inplace
,
Log1pInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
log1p_inplace
,
expected
=
numpy
.
log1p
,
expected
=
numpy
.
log1p
,
good
=
_good_broadcast_unary_positive
,
good
=
_good_broadcast_unary_positive_float
,
grad
=
_grad_broadcast_unary_positive
,
grad
=
_grad_broadcast_unary_positive
,
inplace
=
True
)
inplace
=
True
)
SqrtTester
=
makeBroadcastTester
(
op
=
tensor
.
sqrt
,
SqrtTester
=
makeBroadcastTester
(
op
=
tensor
.
sqrt
,
expected
=
numpy
.
sqrt
,
expected
=
upcast_float16_ufunc
(
numpy
.
sqrt
)
,
good
=
_good_broadcast_unary_positive
,
good
=
_good_broadcast_unary_positive
,
grad
=
_grad_broadcast_unary_positive
)
grad
=
_grad_broadcast_unary_positive
)
SqrtInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
sqrt_inplace
,
SqrtInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
sqrt_inplace
,
expected
=
numpy
.
sqrt
,
expected
=
numpy
.
sqrt
,
good
=
_good_broadcast_unary_positive
,
good
=
_good_broadcast_unary_positive_float
,
grad
=
_grad_broadcast_unary_positive
,
grad
=
_grad_broadcast_unary_positive
,
inplace
=
True
)
inplace
=
True
)
_good_broadcast_unary_wide
=
dict
(
_good_broadcast_unary_wide
=
dict
(
normal
=
(
rand_ranged
(
-
1000
,
1000
,
(
2
,
3
)),),
normal
=
(
rand_ranged
(
-
1000
,
1000
,
(
2
,
3
)),),
integers
=
(
randint_ranged
(
-
1000
,
1000
,
(
2
,
3
)),),
integers
=
(
randint_ranged
(
-
1000
,
1000
,
(
2
,
3
)),),
int8
=
[
numpy
.
arange
(
-
127
,
128
,
dtype
=
'int8'
)],
complex
=
(
randc128_ranged
(
-
1000
,
1000
,
(
2
,
3
)),),
complex
=
(
randc128_ranged
(
-
1000
,
1000
,
(
2
,
3
)),),
empty
=
(
numpy
.
asarray
([],
dtype
=
config
.
floatX
),),)
empty
=
(
numpy
.
asarray
([],
dtype
=
config
.
floatX
),),)
_good_broadcast_unary_wide_float
=
copymod
(
_good_broadcast_unary_wide
,
without
=
[
'integers'
,
'int8'
])
_grad_broadcast_unary_wide
=
dict
(
normal
=
(
rand_ranged
(
-
1000
,
1000
,
(
2
,
3
)),),)
_grad_broadcast_unary_wide
=
dict
(
normal
=
(
rand_ranged
(
-
1000
,
1000
,
(
2
,
3
)),),)
if
theano
.
config
.
floatX
==
'float32'
:
if
theano
.
config
.
floatX
==
'float32'
:
...
@@ -1275,75 +1375,84 @@ else:
...
@@ -1275,75 +1375,84 @@ else:
Deg2radTester
=
makeBroadcastTester
(
Deg2radTester
=
makeBroadcastTester
(
op
=
tensor
.
deg2rad
,
op
=
tensor
.
deg2rad
,
expected
=
numpy
.
deg2rad
,
expected
=
upcast_float16_ufunc
(
numpy
.
deg2rad
)
,
good
=
_good_broadcast_unary_normal_no_complex
,
good
=
_good_broadcast_unary_normal_no_complex
,
grad
=
_grad_broadcast_unary_normal_no_complex
,
grad
=
_grad_broadcast_unary_normal_no_complex
,
eps
=
angle_eps
)
eps
=
angle_eps
)
Deg2radInplaceTester
=
makeBroadcastTester
(
Deg2radInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
deg2rad_inplace
,
op
=
inplace
.
deg2rad_inplace
,
expected
=
numpy
.
deg2rad
,
expected
=
numpy
.
deg2rad
,
good
=
_good_broadcast_unary_normal_no_complex
,
good
=
_good_broadcast_unary_normal_
float_
no_complex
,
grad
=
_grad_broadcast_unary_normal_no_complex
,
grad
=
_grad_broadcast_unary_normal_no_complex
,
inplace
=
True
,
inplace
=
True
,
eps
=
angle_eps
)
eps
=
angle_eps
)
Rad2degTester
=
makeBroadcastTester
(
Rad2degTester
=
makeBroadcastTester
(
op
=
tensor
.
rad2deg
,
op
=
tensor
.
rad2deg
,
expected
=
numpy
.
rad2deg
,
expected
=
upcast_float16_ufunc
(
numpy
.
rad2deg
)
,
good
=
_good_broadcast_unary_normal_no_complex
,
good
=
_good_broadcast_unary_normal_no_complex
,
grad
=
_grad_broadcast_unary_normal_no_complex
,
grad
=
_grad_broadcast_unary_normal_no_complex
,
eps
=
angle_eps
)
eps
=
angle_eps
)
Rad2degInplaceTester
=
makeBroadcastTester
(
Rad2degInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
rad2deg_inplace
,
op
=
inplace
.
rad2deg_inplace
,
expected
=
numpy
.
rad2deg
,
expected
=
numpy
.
rad2deg
,
good
=
_good_broadcast_unary_normal_no_complex
,
good
=
_good_broadcast_unary_normal_
float_
no_complex
,
grad
=
_grad_broadcast_unary_normal_no_complex
,
grad
=
_grad_broadcast_unary_normal_no_complex
,
inplace
=
True
,
inplace
=
True
,
eps
=
angle_eps
)
eps
=
angle_eps
)
SinTester
=
makeBroadcastTester
(
op
=
tensor
.
sin
,
SinTester
=
makeBroadcastTester
(
op
=
tensor
.
sin
,
expected
=
numpy
.
sin
,
expected
=
upcast_float16_ufunc
(
numpy
.
sin
)
,
good
=
_good_broadcast_unary_wide
,
good
=
_good_broadcast_unary_wide
,
grad
=
_grad_broadcast_unary_wide
)
grad
=
_grad_broadcast_unary_wide
)
SinInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
sin_inplace
,
SinInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
sin_inplace
,
expected
=
numpy
.
sin
,
expected
=
numpy
.
sin
,
good
=
_good_broadcast_unary_wide
,
good
=
_good_broadcast_unary_wide_float
,
grad
=
_grad_broadcast_unary_wide
,
grad
=
_grad_broadcast_unary_wide
,
inplace
=
True
)
inplace
=
True
)
_good_broadcast_unary_arcsin
=
dict
(
normal
=
(
rand_ranged
(
-
1
,
1
,
(
2
,
3
)),),
_good_broadcast_unary_arcsin
=
dict
(
normal
=
(
rand_ranged
(
-
1
,
1
,
(
2
,
3
)),),
integers
=
(
randint_ranged
(
-
1
,
1
,
(
2
,
3
)),),
integers
=
(
randint_ranged
(
-
1
,
1
,
(
2
,
3
)),),
int8
=
[
numpy
.
arange
(
-
1
,
2
,
dtype
=
'int8'
)],
complex
=
(
randc128_ranged
(
-
1
,
1
,
(
2
,
3
)),),
complex
=
(
randc128_ranged
(
-
1
,
1
,
(
2
,
3
)),),
empty
=
(
numpy
.
asarray
([],
dtype
=
config
.
floatX
),),)
empty
=
(
numpy
.
asarray
([],
dtype
=
config
.
floatX
),),)
_good_broadcast_unary_arcsin_float
=
copymod
(
_good_broadcast_unary_arcsin
,
without
=
[
'integers'
,
'int8'
])
_grad_broadcast_unary_arcsin
=
dict
(
normal
=
(
rand_ranged
(
-
1
,
1
,
(
2
,
3
)),),)
_grad_broadcast_unary_arcsin
=
dict
(
normal
=
(
rand_ranged
(
-
1
,
1
,
(
2
,
3
)),),)
ArcsinTester
=
makeBroadcastTester
(
op
=
tensor
.
arcsin
,
ArcsinTester
=
makeBroadcastTester
(
op
=
tensor
.
arcsin
,
expected
=
numpy
.
arcsin
,
expected
=
upcast_float16_ufunc
(
numpy
.
arcsin
)
,
good
=
_good_broadcast_unary_arcsin
,
good
=
_good_broadcast_unary_arcsin
,
grad
=
_grad_broadcast_unary_arcsin
)
grad
=
_grad_broadcast_unary_arcsin
)
ArcsinInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
arcsin_inplace
,
ArcsinInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
arcsin_inplace
,
expected
=
numpy
.
arcsin
,
expected
=
numpy
.
arcsin
,
good
=
_good_broadcast_unary_arcsin
,
good
=
_good_broadcast_unary_arcsin_float
,
grad
=
_grad_broadcast_unary_arcsin
,
grad
=
_grad_broadcast_unary_arcsin
,
inplace
=
True
)
inplace
=
True
)
CosTester
=
makeBroadcastTester
(
op
=
tensor
.
cos
,
CosTester
=
makeBroadcastTester
(
op
=
tensor
.
cos
,
expected
=
numpy
.
cos
,
expected
=
upcast_float16_ufunc
(
numpy
.
cos
)
,
good
=
_good_broadcast_unary_wide
,
good
=
_good_broadcast_unary_wide
,
grad
=
_grad_broadcast_unary_wide
)
grad
=
_grad_broadcast_unary_wide
)
CosInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
cos_inplace
,
CosInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
cos_inplace
,
expected
=
numpy
.
cos
,
expected
=
numpy
.
cos
,
good
=
_good_broadcast_unary_wide
,
good
=
_good_broadcast_unary_wide_float
,
grad
=
_grad_broadcast_unary_wide
,
grad
=
_grad_broadcast_unary_wide
,
inplace
=
True
)
inplace
=
True
)
ArccosTester
=
makeBroadcastTester
(
op
=
tensor
.
arccos
,
ArccosTester
=
makeBroadcastTester
(
op
=
tensor
.
arccos
,
expected
=
numpy
.
arccos
,
expected
=
upcast_float16_ufunc
(
numpy
.
arccos
)
,
good
=
_good_broadcast_unary_arcsin
,
good
=
_good_broadcast_unary_arcsin
,
grad
=
_grad_broadcast_unary_arcsin
)
grad
=
_grad_broadcast_unary_arcsin
)
ArccosInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
arccos_inplace
,
ArccosInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
arccos_inplace
,
expected
=
numpy
.
arccos
,
expected
=
numpy
.
arccos
,
good
=
_good_broadcast_unary_arcsin
,
good
=
_good_broadcast_unary_arcsin_float
,
grad
=
_grad_broadcast_unary_arcsin
,
grad
=
_grad_broadcast_unary_arcsin
,
inplace
=
True
)
inplace
=
True
)
...
@@ -1351,6 +1460,7 @@ _good_broadcast_unary_tan = dict(
...
@@ -1351,6 +1460,7 @@ _good_broadcast_unary_tan = dict(
normal
=
(
rand_ranged
(
-
3.14
,
3.14
,
(
2
,
3
)),),
normal
=
(
rand_ranged
(
-
3.14
,
3.14
,
(
2
,
3
)),),
shifted
=
(
rand_ranged
(
3.15
,
6.28
,
(
2
,
3
)),),
shifted
=
(
rand_ranged
(
3.15
,
6.28
,
(
2
,
3
)),),
integers
=
(
randint_ranged
(
-
3
,
3
,
(
2
,
3
)),),
integers
=
(
randint_ranged
(
-
3
,
3
,
(
2
,
3
)),),
int8
=
[
numpy
.
arange
(
-
3
,
4
,
dtype
=
'int8'
)],
complex
=
(
randc128_ranged
(
-
3.14
,
3.14
,
(
2
,
3
)),),
complex
=
(
randc128_ranged
(
-
3.14
,
3.14
,
(
2
,
3
)),),
empty
=
(
numpy
.
asarray
([],
dtype
=
config
.
floatX
),),)
empty
=
(
numpy
.
asarray
([],
dtype
=
config
.
floatX
),),)
#We do not want to test around the discontinuity.
#We do not want to test around the discontinuity.
...
@@ -1358,23 +1468,25 @@ _grad_broadcast_unary_tan = dict(normal=(rand_ranged(-1.5, 1.5, (2, 3)),),
...
@@ -1358,23 +1468,25 @@ _grad_broadcast_unary_tan = dict(normal=(rand_ranged(-1.5, 1.5, (2, 3)),),
shifted
=
(
rand_ranged
(
1.6
,
4.6
,
(
2
,
3
)),))
shifted
=
(
rand_ranged
(
1.6
,
4.6
,
(
2
,
3
)),))
TanTester
=
makeBroadcastTester
(
op
=
tensor
.
tan
,
TanTester
=
makeBroadcastTester
(
op
=
tensor
.
tan
,
expected
=
numpy
.
tan
,
expected
=
upcast_float16_ufunc
(
numpy
.
tan
)
,
good
=
_good_broadcast_unary_tan
,
good
=
_good_broadcast_unary_tan
,
grad
=
_grad_broadcast_unary_tan
)
grad
=
_grad_broadcast_unary_tan
)
TanInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
tan_inplace
,
TanInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
tan_inplace
,
expected
=
numpy
.
tan
,
expected
=
numpy
.
tan
,
good
=
_good_broadcast_unary_tan
,
good
=
copymod
(
_good_broadcast_unary_tan
,
without
=
[
'integers'
,
'int8'
])
,
grad
=
_grad_broadcast_unary_tan
,
grad
=
_grad_broadcast_unary_tan
,
inplace
=
True
)
inplace
=
True
)
ArctanTester
=
makeBroadcastTester
(
op
=
tensor
.
arctan
,
ArctanTester
=
makeBroadcastTester
(
op
=
tensor
.
arctan
,
expected
=
numpy
.
arctan
,
expected
=
upcast_float16_ufunc
(
numpy
.
arctan
)
,
good
=
_good_broadcast_unary_wide
,
good
=
_good_broadcast_unary_wide
,
grad
=
_grad_broadcast_unary_wide
)
grad
=
_grad_broadcast_unary_wide
)
ArctanInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
arctan_inplace
,
ArctanInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
arctan_inplace
,
expected
=
numpy
.
arctan
,
expected
=
numpy
.
arctan
,
good
=
_good_broadcast_unary_wide
,
good
=
_good_broadcast_unary_wide_float
,
grad
=
_grad_broadcast_unary_wide
,
grad
=
_grad_broadcast_unary_wide
,
inplace
=
True
)
inplace
=
True
)
...
@@ -1385,6 +1497,8 @@ _good_broadcast_binary_arctan2 = dict(
...
@@ -1385,6 +1497,8 @@ _good_broadcast_binary_arctan2 = dict(
row
=
(
rand
(
2
,
3
),
rand
(
1
,
3
)),
row
=
(
rand
(
2
,
3
),
rand
(
1
,
3
)),
column
=
(
rand
(
2
,
3
),
rand
(
2
,
1
)),
column
=
(
rand
(
2
,
3
),
rand
(
2
,
1
)),
integers
=
(
randint
(
2
,
3
),
randint
(
2
,
3
)),
integers
=
(
randint
(
2
,
3
),
randint
(
2
,
3
)),
int8
=
[
numpy
.
arange
(
-
127
,
128
,
dtype
=
'int8'
),
numpy
.
arange
(
-
127
,
128
,
dtype
=
'int8'
)[:,
numpy
.
newaxis
]],
dtype_mixup_1
=
(
rand
(
2
,
3
),
randint
(
2
,
3
)),
dtype_mixup_1
=
(
rand
(
2
,
3
),
randint
(
2
,
3
)),
dtype_mixup_2
=
(
randint
(
2
,
3
),
rand
(
2
,
3
)),
dtype_mixup_2
=
(
randint
(
2
,
3
),
rand
(
2
,
3
)),
empty
=
(
numpy
.
asarray
([],
dtype
=
config
.
floatX
),
empty
=
(
numpy
.
asarray
([],
dtype
=
config
.
floatX
),
...
@@ -1398,70 +1512,84 @@ _grad_broadcast_binary_arctan2 = dict(
...
@@ -1398,70 +1512,84 @@ _grad_broadcast_binary_arctan2 = dict(
column
=
(
rand
(
2
,
3
),
rand
(
2
,
1
)),
column
=
(
rand
(
2
,
3
),
rand
(
2
,
1
)),
)
)
Arctan2Tester
=
makeBroadcastTester
(
op
=
tensor
.
arctan2
,
Arctan2Tester
=
makeBroadcastTester
(
expected
=
numpy
.
arctan2
,
op
=
tensor
.
arctan2
,
expected
=
upcast_float16_ufunc
(
numpy
.
arctan2
),
good
=
_good_broadcast_binary_arctan2
,
good
=
_good_broadcast_binary_arctan2
,
grad
=
_grad_broadcast_binary_arctan2
)
grad
=
_grad_broadcast_binary_arctan2
)
Arctan2InplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
arctan2_inplace
,
Arctan2InplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
arctan2_inplace
,
expected
=
numpy
.
arctan2
,
expected
=
numpy
.
arctan2
,
good
=
_good_broadcast_binary_arctan2
,
good
=
copymod
(
_good_broadcast_binary_arctan2
,
without
=
[
'integers'
,
'int8'
])
,
grad
=
_grad_broadcast_binary_arctan2
,
grad
=
_grad_broadcast_binary_arctan2
,
inplace
=
True
)
inplace
=
True
)
CoshTester
=
makeBroadcastTester
(
op
=
tensor
.
cosh
,
CoshTester
=
makeBroadcastTester
(
expected
=
numpy
.
cosh
,
op
=
tensor
.
cosh
,
good
=
_good_broadcast_unary_normal
,
expected
=
upcast_float16_ufunc
(
numpy
.
cosh
),
good
=
dict
(
_good_broadcast_unary_normal
,
int8
=
[
numpy
.
arange
(
-
89
,
90
,
dtype
=
'int8'
)]),
grad
=
_grad_broadcast_unary_normal
)
grad
=
_grad_broadcast_unary_normal
)
CoshInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
cosh_inplace
,
CoshInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
cosh_inplace
,
expected
=
numpy
.
cosh
,
expected
=
numpy
.
cosh
,
good
=
_good_broadcast_unary_normal
,
good
=
_good_broadcast_unary_normal_float
,
grad
=
_grad_broadcast_unary_normal
,
grad
=
_grad_broadcast_unary_normal
,
inplace
=
True
)
inplace
=
True
)
_good_broadcast_unary_arccosh
=
dict
(
_good_broadcast_unary_arccosh
=
dict
(
normal
=
(
rand_ranged
(
1
,
1000
,
(
2
,
3
)),),
normal
=
(
rand_ranged
(
1
,
1000
,
(
2
,
3
)),),
integers
=
(
randint_ranged
(
1
,
1000
,
(
2
,
3
)),),
integers
=
(
randint_ranged
(
1
,
1000
,
(
2
,
3
)),),
uint8
=
[
numpy
.
arange
(
1
,
256
,
dtype
=
'uint8'
)],
complex
=
(
randc128_ranged
(
1
,
1000
,
(
2
,
3
)),),
complex
=
(
randc128_ranged
(
1
,
1000
,
(
2
,
3
)),),
empty
=
(
numpy
.
asarray
([],
dtype
=
config
.
floatX
),),)
empty
=
(
numpy
.
asarray
([],
dtype
=
config
.
floatX
),),)
_grad_broadcast_unary_arccosh
=
dict
(
normal
=
(
rand_ranged
(
1
,
1000
,
(
2
,
3
)),),)
_grad_broadcast_unary_arccosh
=
dict
(
normal
=
(
rand_ranged
(
1
,
1000
,
(
2
,
3
)),),)
ArccoshTester
=
makeBroadcastTester
(
op
=
tensor
.
arccosh
,
ArccoshTester
=
makeBroadcastTester
(
expected
=
numpy
.
arccosh
,
op
=
tensor
.
arccosh
,
expected
=
upcast_float16_ufunc
(
numpy
.
arccosh
),
good
=
_good_broadcast_unary_arccosh
,
good
=
_good_broadcast_unary_arccosh
,
grad
=
_grad_broadcast_unary_arccosh
)
grad
=
_grad_broadcast_unary_arccosh
)
ArccoshInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
arccosh_inplace
,
ArccoshInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
arccosh_inplace
,
expected
=
numpy
.
arccosh
,
expected
=
numpy
.
arccosh
,
good
=
_good_broadcast_unary_arccosh
,
good
=
copymod
(
_good_broadcast_unary_arccosh
,
without
=
[
'integers'
,
'uint8'
])
,
grad
=
_grad_broadcast_unary_arccosh
,
grad
=
_grad_broadcast_unary_arccosh
,
inplace
=
True
)
inplace
=
True
)
SinhTester
=
makeBroadcastTester
(
op
=
tensor
.
sinh
,
SinhTester
=
makeBroadcastTester
(
expected
=
numpy
.
sinh
,
op
=
tensor
.
sinh
,
good
=
_good_broadcast_unary_normal
,
expected
=
upcast_float16_ufunc
(
numpy
.
sinh
),
good
=
dict
(
_good_broadcast_unary_normal
,
int8
=
[
numpy
.
arange
(
-
89
,
90
,
dtype
=
'int8'
)]),
grad
=
_grad_broadcast_unary_normal
)
grad
=
_grad_broadcast_unary_normal
)
SinhInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
sinh_inplace
,
SinhInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
sinh_inplace
,
expected
=
numpy
.
sinh
,
expected
=
numpy
.
sinh
,
good
=
_good_broadcast_unary_normal
,
good
=
_good_broadcast_unary_normal_float
,
grad
=
_grad_broadcast_unary_normal
,
grad
=
_grad_broadcast_unary_normal
,
inplace
=
True
)
inplace
=
True
)
ArcsinhTester
=
makeBroadcastTester
(
op
=
tensor
.
arcsinh
,
ArcsinhTester
=
makeBroadcastTester
(
expected
=
numpy
.
arcsinh
,
op
=
tensor
.
arcsinh
,
expected
=
upcast_float16_ufunc
(
numpy
.
arcsinh
),
good
=
_good_broadcast_unary_normal
,
good
=
_good_broadcast_unary_normal
,
grad
=
_grad_broadcast_unary_normal
)
grad
=
_grad_broadcast_unary_normal
)
ArcsinhInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
arcsinh_inplace
,
ArcsinhInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
arcsinh_inplace
,
expected
=
numpy
.
arcsinh
,
expected
=
numpy
.
arcsinh
,
good
=
_good_broadcast_unary_normal
,
good
=
_good_broadcast_unary_normal_float
,
grad
=
_grad_broadcast_unary_normal
,
grad
=
_grad_broadcast_unary_normal
,
inplace
=
True
)
inplace
=
True
)
TanhTester
=
makeBroadcastTester
(
op
=
tensor
.
tanh
,
TanhTester
=
makeBroadcastTester
(
op
=
tensor
.
tanh
,
expected
=
numpy
.
tanh
,
expected
=
upcast_float16_ufunc
(
numpy
.
tanh
)
,
good
=
_good_broadcast_unary_normal
,
good
=
_good_broadcast_unary_normal
,
grad
=
_grad_broadcast_unary_normal
)
grad
=
_grad_broadcast_unary_normal
)
TanhInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
tanh_inplace
,
TanhInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
tanh_inplace
,
expected
=
numpy
.
tanh
,
expected
=
numpy
.
tanh
,
good
=
_good_broadcast_unary_normal
,
good
=
_good_broadcast_unary_normal_float
,
grad
=
_grad_broadcast_unary_normal
,
grad
=
_grad_broadcast_unary_normal
,
inplace
=
True
)
inplace
=
True
)
...
@@ -1469,29 +1597,25 @@ _eps = 1e-10
...
@@ -1469,29 +1597,25 @@ _eps = 1e-10
_good_broadcast_unary_arctanh
=
dict
(
_good_broadcast_unary_arctanh
=
dict
(
normal
=
(
rand_ranged
(
-
1
+
_eps
,
1
-
_eps
,
(
2
,
3
)),),
normal
=
(
rand_ranged
(
-
1
+
_eps
,
1
-
_eps
,
(
2
,
3
)),),
integers
=
(
randint_ranged
(
-
1
+
_eps
,
1
-
_eps
,
(
2
,
3
)),),
integers
=
(
randint_ranged
(
-
1
+
_eps
,
1
-
_eps
,
(
2
,
3
)),),
int8
=
[
numpy
.
arange
(
0
,
1
,
dtype
=
'int8'
)],
complex
=
(
randc128_ranged
(
-
1
+
_eps
,
1
-
_eps
,
(
2
,
3
)),),
complex
=
(
randc128_ranged
(
-
1
+
_eps
,
1
-
_eps
,
(
2
,
3
)),),
empty
=
(
numpy
.
asarray
([],
dtype
=
config
.
floatX
),),)
empty
=
(
numpy
.
asarray
([],
dtype
=
config
.
floatX
),),)
_grad_broadcast_unary_arctanh
=
dict
(
_grad_broadcast_unary_arctanh
=
dict
(
normal
=
(
rand_ranged
(
-
1
+
_eps
,
1
-
_eps
,
(
2
,
3
)),),)
normal
=
(
rand_ranged
(
-
1
+
_eps
,
1
-
_eps
,
(
2
,
3
)),),)
ArctanhTester
=
makeBroadcastTester
(
op
=
tensor
.
arctanh
,
ArctanhTester
=
makeBroadcastTester
(
expected
=
numpy
.
arctanh
,
op
=
tensor
.
arctanh
,
expected
=
upcast_float16_ufunc
(
numpy
.
arctanh
),
good
=
_good_broadcast_unary_arctanh
,
good
=
_good_broadcast_unary_arctanh
,
grad
=
_grad_broadcast_unary_arctanh
)
grad
=
_grad_broadcast_unary_arctanh
)
ArctanhInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
arctanh_inplace
,
ArctanhInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
arctanh_inplace
,
expected
=
numpy
.
arctanh
,
expected
=
numpy
.
arctanh
,
good
=
_good_broadcast_unary_arctanh
,
good
=
copymod
(
_good_broadcast_unary_arctanh
,
without
=
[
'integers'
,
'int8'
])
,
grad
=
_grad_broadcast_unary_arctanh
,
grad
=
_grad_broadcast_unary_arctanh
,
inplace
=
True
)
inplace
=
True
)
#inplace ops when the input is integer and the output is float*
# don't have a well defined behavior. We don't test that case.
_good_broadcast_unary_normal_no_int_no_complex
=
_good_broadcast_unary_normal_no_complex
.
copy
()
del
_good_broadcast_unary_normal_no_int_no_complex
[
'integers'
]
_good_broadcast_unary_normal_no_int
=
_good_broadcast_unary_normal
.
copy
()
del
_good_broadcast_unary_normal_no_int
[
'integers'
]
# We can't test it if scipy is not installed!
# We can't test it if scipy is not installed!
# Precomputing the result is brittle(it have been broken!)
# Precomputing the result is brittle(it have been broken!)
# As if we do any modification to random number here,
# As if we do any modification to random number here,
...
@@ -1528,7 +1652,7 @@ ErfTester = makeBroadcastTester(
...
@@ -1528,7 +1652,7 @@ ErfTester = makeBroadcastTester(
ErfInplaceTester
=
makeBroadcastTester
(
ErfInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
erf_inplace
,
op
=
inplace
.
erf_inplace
,
expected
=
expected_erf
,
expected
=
expected_erf
,
good
=
_good_broadcast_unary_normal_
no_in
t
,
good
=
_good_broadcast_unary_normal_
floa
t
,
grad
=
_grad_broadcast_unary_normal
,
grad
=
_grad_broadcast_unary_normal
,
mode
=
mode_no_scipy
,
mode
=
mode_no_scipy
,
eps
=
2e-10
,
eps
=
2e-10
,
...
@@ -1538,7 +1662,7 @@ ErfInplaceTester = makeBroadcastTester(
...
@@ -1538,7 +1662,7 @@ ErfInplaceTester = makeBroadcastTester(
ErfcTester
=
makeBroadcastTester
(
ErfcTester
=
makeBroadcastTester
(
op
=
tensor
.
erfc
,
op
=
tensor
.
erfc
,
expected
=
expected_erfc
,
expected
=
expected_erfc
,
good
=
_good_broadcast_unary_normal_
no_in
t_no_complex
,
good
=
_good_broadcast_unary_normal_
floa
t_no_complex
,
grad
=
_grad_broadcast_unary_normal
,
grad
=
_grad_broadcast_unary_normal
,
eps
=
2e-10
,
eps
=
2e-10
,
mode
=
mode_no_scipy
,
mode
=
mode_no_scipy
,
...
@@ -1546,7 +1670,7 @@ ErfcTester = makeBroadcastTester(
...
@@ -1546,7 +1670,7 @@ ErfcTester = makeBroadcastTester(
ErfcInplaceTester
=
makeBroadcastTester
(
ErfcInplaceTester
=
makeBroadcastTester
(
op
=
inplace
.
erfc_inplace
,
op
=
inplace
.
erfc_inplace
,
expected
=
expected_erfc
,
expected
=
expected_erfc
,
good
=
_good_broadcast_unary_normal_
no_in
t_no_complex
,
good
=
_good_broadcast_unary_normal_
floa
t_no_complex
,
grad
=
_grad_broadcast_unary_normal
,
grad
=
_grad_broadcast_unary_normal
,
eps
=
2e-10
,
eps
=
2e-10
,
mode
=
mode_no_scipy
,
mode
=
mode_no_scipy
,
...
@@ -1556,7 +1680,7 @@ ErfcInplaceTester = makeBroadcastTester(
...
@@ -1556,7 +1680,7 @@ ErfcInplaceTester = makeBroadcastTester(
ErfinvTester
=
makeBroadcastTester
(
ErfinvTester
=
makeBroadcastTester
(
op
=
tensor
.
erfinv
,
op
=
tensor
.
erfinv
,
expected
=
expected_erfinv
,
expected
=
expected_erfinv
,
good
=
_good_broadcast_unary_normal_
no_in
t_no_complex
,
good
=
_good_broadcast_unary_normal_
floa
t_no_complex
,
grad
=
_grad_broadcast_unary_abs1_no_complex
,
grad
=
_grad_broadcast_unary_abs1_no_complex
,
eps
=
2e-10
,
eps
=
2e-10
,
mode
=
mode_no_scipy
,
mode
=
mode_no_scipy
,
...
@@ -1565,7 +1689,7 @@ ErfinvTester = makeBroadcastTester(
...
@@ -1565,7 +1689,7 @@ ErfinvTester = makeBroadcastTester(
ErfcinvTester
=
makeBroadcastTester
(
ErfcinvTester
=
makeBroadcastTester
(
op
=
tensor
.
erfcinv
,
op
=
tensor
.
erfcinv
,
expected
=
expected_erfcinv
,
expected
=
expected_erfcinv
,
good
=
_good_broadcast_unary_normal_
no_in
t_no_complex
,
good
=
_good_broadcast_unary_normal_
floa
t_no_complex
,
grad
=
_grad_broadcast_unary_0_2_no_complex
,
grad
=
_grad_broadcast_unary_0_2_no_complex
,
eps
=
2e-10
,
eps
=
2e-10
,
mode
=
mode_no_scipy
,
mode
=
mode_no_scipy
,
...
...
编写
预览
Markdown
格式
0%
重试
或
添加新文件
添加附件
取消
您添加了
0
人
到此讨论。请谨慎行事。
请先完成此评论的编辑!
取消
请
注册
或者
登录
后发表评论