提交 7c0a5fa0 authored 作者: Nicolas Bouchard's avatar Nicolas Bouchard

Add new elemwise.

上级 4cc35522
......@@ -3280,6 +3280,34 @@ def log1p(x):
# see decorator for function body
@structured_monoid(tensor.expm1)
def expm1(x):
"""Elemwise e^`x` - 1.
"""
# see decorator for function body
@structured_monoid(tensor.deg2rad)
def deg2rad(x):
"""Elemwise degree to radian.
"""
# see decorator for function body
@structured_monoid(tensor.rad2deg)
def rad2deg(x):
"""Elemwise radian to degree.
"""
# see decorator for function body
@structured_monoid(tensor.trunc)
def trunc(x):
"""Elemwise truncature.
"""
# see decorator for function body
@structured_monoid(tensor.sqr)
def sqr(x):
"""Elemwise `x` * `x`.
......
......@@ -2490,6 +2490,30 @@ Log1pTester = elemwise_checker(
numpy.log1p,
gap=(0.5, 10))
Expm1Tester = elemwise_checker(
sparse.expm1,
numpy.expm1)
Deg2RadTester = elemwise_checker(
sparse.deg2rad,
numpy.deg2rad,
test_dtypes=[m for m in sparse.all_dtypes
if not m in sparse.complex_dtypes])
Rad2DegTester = elemwise_checker(
sparse.rad2deg,
numpy.rad2deg,
test_dtypes=[m for m in sparse.all_dtypes
if not m in sparse.complex_dtypes])
TruncTester = elemwise_checker(
sparse.trunc,
numpy.trunc,
test_dtypes=[m for m in sparse.all_dtypes
if not m in sparse.complex_dtypes])
SqrTester = elemwise_checker(
sparse.sqr,
lambda x: x * x)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论