提交 88dfd88f authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Rename _scal_elemwise to scalar_elemwise

上级 e701d8f8
差异被折叠。
...@@ -2242,7 +2242,7 @@ class ProdWithoutZeros(CAReduceDtype): ...@@ -2242,7 +2242,7 @@ class ProdWithoutZeros(CAReduceDtype):
return [a_grad] return [a_grad]
def _scal_elemwise(*symbol, nfunc=None, nin=None, nout=None, symbolname=None): def scalar_elemwise(*symbol, nfunc=None, nin=None, nout=None, symbolname=None):
"""Replace a symbol definition with an `Elemwise`-wrapped version of the corresponding scalar `Op`. """Replace a symbol definition with an `Elemwise`-wrapped version of the corresponding scalar `Op`.
If it is not ``None``, the `nfunc` argument should be a string such that If it is not ``None``, the `nfunc` argument should be a string such that
......
from theano import printing from theano import printing
from theano.printing import pprint from theano.printing import pprint
from theano.tensor.elemwise import DimShuffle, _scal_elemwise from theano.tensor.elemwise import DimShuffle, scalar_elemwise
@_scal_elemwise @scalar_elemwise
def lt_inplace(a, b): def lt_inplace(a, b):
"""a < b (inplace on a)""" """a < b (inplace on a)"""
@_scal_elemwise @scalar_elemwise
def gt_inplace(a, b): def gt_inplace(a, b):
"""a > b (inplace on a)""" """a > b (inplace on a)"""
@_scal_elemwise @scalar_elemwise
def le_inplace(a, b): def le_inplace(a, b):
"""a <= b (inplace on a)""" """a <= b (inplace on a)"""
@_scal_elemwise @scalar_elemwise
def ge_inplace(a, b): def ge_inplace(a, b):
"""a >= b (inplace on a)""" """a >= b (inplace on a)"""
@_scal_elemwise @scalar_elemwise
def eq_inplace(a, b): def eq_inplace(a, b):
"""a == b (inplace on a)""" """a == b (inplace on a)"""
@_scal_elemwise @scalar_elemwise
def neq_inplace(a, b): def neq_inplace(a, b):
"""a != b (inplace on a)""" """a != b (inplace on a)"""
@_scal_elemwise @scalar_elemwise
def and__inplace(a, b): def and__inplace(a, b):
"""bitwise a & b (inplace on a)""" """bitwise a & b (inplace on a)"""
@_scal_elemwise @scalar_elemwise
def or__inplace(a, b): def or__inplace(a, b):
"""bitwise a | b (inplace on a)""" """bitwise a | b (inplace on a)"""
@_scal_elemwise @scalar_elemwise
def xor_inplace(a, b): def xor_inplace(a, b):
"""bitwise a ^ b (inplace on a)""" """bitwise a ^ b (inplace on a)"""
@_scal_elemwise @scalar_elemwise
def invert_inplace(a): def invert_inplace(a):
"""bitwise ~a (inplace on a)""" """bitwise ~a (inplace on a)"""
@_scal_elemwise @scalar_elemwise
def abs__inplace(a): def abs__inplace(a):
"""|`a`| (inplace on `a`)""" """|`a`| (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def exp_inplace(a): def exp_inplace(a):
"""e^`a` (inplace on `a`)""" """e^`a` (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def exp2_inplace(a): def exp2_inplace(a):
"""2^`a` (inplace on `a`)""" """2^`a` (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def expm1_inplace(a): def expm1_inplace(a):
"""e^`a` - 1 (inplace on `a`)""" """e^`a` - 1 (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def neg_inplace(a): def neg_inplace(a):
"""-a (inplace on a)""" """-a (inplace on a)"""
@_scal_elemwise @scalar_elemwise
def inv_inplace(a): def inv_inplace(a):
"""1.0/a (inplace on a)""" """1.0/a (inplace on a)"""
@_scal_elemwise @scalar_elemwise
def log_inplace(a): def log_inplace(a):
"""base e logarithm of a (inplace on a)""" """base e logarithm of a (inplace on a)"""
@_scal_elemwise @scalar_elemwise
def log1p_inplace(a): def log1p_inplace(a):
"""log(1+a)""" """log(1+a)"""
@_scal_elemwise @scalar_elemwise
def log2_inplace(a): def log2_inplace(a):
"""base 2 logarithm of a (inplace on a)""" """base 2 logarithm of a (inplace on a)"""
@_scal_elemwise @scalar_elemwise
def log10_inplace(a): def log10_inplace(a):
"""base 10 logarithm of a (inplace on a)""" """base 10 logarithm of a (inplace on a)"""
@_scal_elemwise @scalar_elemwise
def sgn_inplace(a): def sgn_inplace(a):
"""sign of `a` (inplace on `a`)""" """sign of `a` (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def ceil_inplace(a): def ceil_inplace(a):
"""ceil of `a` (inplace on `a`)""" """ceil of `a` (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def floor_inplace(a): def floor_inplace(a):
"""floor of `a` (inplace on `a`)""" """floor of `a` (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def trunc_inplace(a): def trunc_inplace(a):
"""trunc of `a` (inplace on `a`)""" """trunc of `a` (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def round_half_to_even_inplace(a): def round_half_to_even_inplace(a):
"""round_half_to_even_inplace(a) (inplace on `a`)""" """round_half_to_even_inplace(a) (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def round_half_away_from_zero_inplace(a): def round_half_away_from_zero_inplace(a):
"""round_half_away_from_zero_inplace(a) (inplace on `a`)""" """round_half_away_from_zero_inplace(a) (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def sqr_inplace(a): def sqr_inplace(a):
"""square of `a` (inplace on `a`)""" """square of `a` (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def sqrt_inplace(a): def sqrt_inplace(a):
"""square root of `a` (inplace on `a`)""" """square root of `a` (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def deg2rad_inplace(a): def deg2rad_inplace(a):
"""convert degree `a` to radian(inplace on `a`)""" """convert degree `a` to radian(inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def rad2deg_inplace(a): def rad2deg_inplace(a):
"""convert radian `a` to degree(inplace on `a`)""" """convert radian `a` to degree(inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def cos_inplace(a): def cos_inplace(a):
"""cosine of `a` (inplace on `a`)""" """cosine of `a` (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def arccos_inplace(a): def arccos_inplace(a):
"""arccosine of `a` (inplace on `a`)""" """arccosine of `a` (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def sin_inplace(a): def sin_inplace(a):
"""sine of `a` (inplace on `a`)""" """sine of `a` (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def arcsin_inplace(a): def arcsin_inplace(a):
"""arcsine of `a` (inplace on `a`)""" """arcsine of `a` (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def tan_inplace(a): def tan_inplace(a):
"""tangent of `a` (inplace on `a`)""" """tangent of `a` (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def arctan_inplace(a): def arctan_inplace(a):
"""arctangent of `a` (inplace on `a`)""" """arctangent of `a` (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def arctan2_inplace(a, b): def arctan2_inplace(a, b):
"""arctangent of `a` / `b` (inplace on `a`)""" """arctangent of `a` / `b` (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def cosh_inplace(a): def cosh_inplace(a):
"""hyperbolic cosine of `a` (inplace on `a`)""" """hyperbolic cosine of `a` (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def arccosh_inplace(a): def arccosh_inplace(a):
"""hyperbolic arc cosine of `a` (inplace on `a`)""" """hyperbolic arc cosine of `a` (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def sinh_inplace(a): def sinh_inplace(a):
"""hyperbolic sine of `a` (inplace on `a`)""" """hyperbolic sine of `a` (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def arcsinh_inplace(a): def arcsinh_inplace(a):
"""hyperbolic arc sine of `a` (inplace on `a`)""" """hyperbolic arc sine of `a` (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def tanh_inplace(a): def tanh_inplace(a):
"""hyperbolic tangent of `a` (inplace on `a`)""" """hyperbolic tangent of `a` (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def arctanh_inplace(a): def arctanh_inplace(a):
"""hyperbolic arc tangent of `a` (inplace on `a`)""" """hyperbolic arc tangent of `a` (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def erf_inplace(a): def erf_inplace(a):
"""error function""" """error function"""
@_scal_elemwise @scalar_elemwise
def erfc_inplace(a): def erfc_inplace(a):
"""complementary error function""" """complementary error function"""
@_scal_elemwise @scalar_elemwise
def erfcx_inplace(a): def erfcx_inplace(a):
"""scaled complementary error function""" """scaled complementary error function"""
@_scal_elemwise @scalar_elemwise
def gamma_inplace(a): def gamma_inplace(a):
"""gamma function""" """gamma function"""
@_scal_elemwise @scalar_elemwise
def gammaln_inplace(a): def gammaln_inplace(a):
"""log gamma function""" """log gamma function"""
@_scal_elemwise @scalar_elemwise
def psi_inplace(a): def psi_inplace(a):
"""derivative of log gamma function""" """derivative of log gamma function"""
@_scal_elemwise @scalar_elemwise
def tri_gamma_inplace(a): def tri_gamma_inplace(a):
"""second derivative of the log gamma function""" """second derivative of the log gamma function"""
@_scal_elemwise @scalar_elemwise
def chi2sf_inplace(x, k): def chi2sf_inplace(x, k):
"""chi squared survival function""" """chi squared survival function"""
@_scal_elemwise @scalar_elemwise
def j0_inplace(x): def j0_inplace(x):
"""Bessel function of the first kind of order 0.""" """Bessel function of the first kind of order 0."""
@_scal_elemwise @scalar_elemwise
def j1_inplace(x): def j1_inplace(x):
"""Bessel function of the first kind of order 1.""" """Bessel function of the first kind of order 1."""
@_scal_elemwise @scalar_elemwise
def jv_inplace(v, x): def jv_inplace(v, x):
"""Bessel function of the first kind of order v (real).""" """Bessel function of the first kind of order v (real)."""
@_scal_elemwise @scalar_elemwise
def i0_inplace(x): def i0_inplace(x):
"""Modified Bessel function of the first kind of order 0.""" """Modified Bessel function of the first kind of order 0."""
@_scal_elemwise @scalar_elemwise
def i1_inplace(x): def i1_inplace(x):
"""Modified Bessel function of the first kind of order 1.""" """Modified Bessel function of the first kind of order 1."""
@_scal_elemwise @scalar_elemwise
def iv_inplace(v, x): def iv_inplace(v, x):
"""Modified Bessel function of the first kind of order v (real).""" """Modified Bessel function of the first kind of order v (real)."""
@_scal_elemwise @scalar_elemwise
def second_inplace(a): def second_inplace(a):
"""Fill `a` with `b`""" """Fill `a` with `b`"""
...@@ -297,52 +297,52 @@ fill_inplace = second_inplace ...@@ -297,52 +297,52 @@ fill_inplace = second_inplace
pprint.assign(fill_inplace, printing.FunctionPrinter("fill=")) pprint.assign(fill_inplace, printing.FunctionPrinter("fill="))
@_scal_elemwise(symbolname="scalar_maximum_inplace") @scalar_elemwise(symbolname="scalar_maximum_inplace")
def maximum_inplace(a, b): def maximum_inplace(a, b):
"""elementwise addition (inplace on `a`)""" """elementwise addition (inplace on `a`)"""
@_scal_elemwise(symbolname="scalar_minimum_inplace") @scalar_elemwise(symbolname="scalar_minimum_inplace")
def minimum_inplace(a, b): def minimum_inplace(a, b):
"""elementwise addition (inplace on `a`)""" """elementwise addition (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def add_inplace(a, b): def add_inplace(a, b):
"""elementwise addition (inplace on `a`)""" """elementwise addition (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def sub_inplace(a, b): def sub_inplace(a, b):
"""elementwise subtraction (inplace on `a`)""" """elementwise subtraction (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def mul_inplace(a, b): def mul_inplace(a, b):
"""elementwise multiplication (inplace on `a`)""" """elementwise multiplication (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def true_div_inplace(a, b): def true_div_inplace(a, b):
"""elementwise division (inplace on `a`)""" """elementwise division (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def int_div_inplace(a, b): def int_div_inplace(a, b):
"""elementwise division (inplace on `a`)""" """elementwise division (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def mod_inplace(a, b): def mod_inplace(a, b):
"""elementwise modulo (inplace on `a`)""" """elementwise modulo (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def pow_inplace(a, b): def pow_inplace(a, b):
"""elementwise power (inplace on `a`)""" """elementwise power (inplace on `a`)"""
@_scal_elemwise @scalar_elemwise
def conj_inplace(a): def conj_inplace(a):
"""elementwise conjugate (inplace on `a`)""" """elementwise conjugate (inplace on `a`)"""
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论