提交 d3613582 authored 作者: projects@lgcm's avatar projects@lgcm

trying doc strategy

上级 cd10efcc
...@@ -490,7 +490,7 @@ elemwise.TensorValue = TensorValue ...@@ -490,7 +490,7 @@ elemwise.TensorValue = TensorValue
# Utilities # Utilities
######################### #########################
def _elemwise(scalar_op, name): def _elemwise(scalar_op, name, doc_prefix=''):
straight = elemwise.Elemwise(scalar_op, name = name) straight = elemwise.Elemwise(scalar_op, name = name)
inplace_scalar_op = scalar_op.__class__(scal.transfer_type(0)) inplace_scalar_op = scalar_op.__class__(scal.transfer_type(0))
inplace = elemwise.Elemwise(inplace_scalar_op, {0: 0}, name = name+"_inplace") inplace = elemwise.Elemwise(inplace_scalar_op, {0: 0}, name = name+"_inplace")
...@@ -503,6 +503,9 @@ def _elemwise(scalar_op, name): ...@@ -503,6 +503,9 @@ def _elemwise(scalar_op, name):
straight.__module__ = 'tensor' straight.__module__ = 'tensor'
inplace.__module__ = 'tensor' inplace.__module__ = 'tensor'
if doc_prefix:
straight.__doc__ = doc_prefix + '\n' + straight.__doc__
return straight, inplace return straight, inplace
...@@ -700,20 +703,20 @@ invert, invert_inplace = _elemwise(scal.invert, 'invert') ...@@ -700,20 +703,20 @@ invert, invert_inplace = _elemwise(scal.invert, 'invert')
# Math # Math
########################## ##########################
_abs, abs_inplace = _elemwise(scal.abs, 'abs') _abs, abs_inplace = _elemwise(scal.abs, 'abs',
"""absolute value (elemwise)""" """absolute value (elemwise)""")
exp, exp_inplace = _elemwise(scal.exp, 'exp') exp, exp_inplace = _elemwise(scal.exp, 'exp',
"""exponential (elemwise)""" """exponential (elemwise)""")
neg, neg_inplace = _elemwise(scal.neg, 'neg') neg, neg_inplace = _elemwise(scal.neg, 'neg',
"""negative (elemwise)""" """negative (elemwise)""")
inv, inv_inplace = _elemwise(scal.inv, 'inv') inv, inv_inplace = _elemwise(scal.inv, 'inv',
"""multiplicative inverse (elemwise)""" """multiplicative inverse (elemwise)""")
log, log_inplace = _elemwise(scal.log, 'log') log, log_inplace = _elemwise(scal.log, 'log',
"""logarithm base-e (elemwise)""" """logarithm base-e (elemwise)""")
log2, log2_inplace = _elemwise(scal.log2, 'log2') log2, log2_inplace = _elemwise(scal.log2, 'log2')
"""logarithm base-2 (elemwise)""" """logarithm base-2 (elemwise)"""
...@@ -867,8 +870,7 @@ repeat = Repeat() ...@@ -867,8 +870,7 @@ repeat = Repeat()
# Arithmetics # Arithmetics
########################## ##########################
add, add_inplace = _elemwise(scal.add, 'add') add, add_inplace = _elemwise(scal.add, 'add', 'addition (element-wise)')
"""addition (elemwise)"""
sub, sub_inplace = _elemwise(scal.sub, 'sub') sub, sub_inplace = _elemwise(scal.sub, 'sub')
"""subtraction (elemwise)""" """subtraction (elemwise)"""
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论