Updated auto-generated operator modules using the fixed fetch_ops
This commit is contained in:
Родитель
ef2a6045af
Коммит
1f0ac9c2f1
|
@ -60,7 +60,6 @@ class If(ComputationNode):
|
|||
self.thenVal = thenVal
|
||||
self.elseVal = elseVal
|
||||
self.params_with_defaults = []
|
||||
self.inputs = ['cond', 'thenVal', 'elseVal']
|
||||
|
||||
class Sign(ComputationNode):
|
||||
def __init__(self, x, op_name='Sign', name=None):
|
||||
|
@ -635,9 +634,9 @@ class RectifiedLinear(ComputationNode):
|
|||
self.params_with_defaults = []
|
||||
self.inputs = ['z']
|
||||
|
||||
class ReducePlus(ComputationNode):
|
||||
def __init__(self, z, axis=0, op_name='ReducePlus', name=None):
|
||||
super(ReducePlus, self).__init__(params=['z', 'axis'], op_name=op_name, name=name)
|
||||
class ReduceSum(ComputationNode):
|
||||
def __init__(self, z, axis=0, op_name='ReduceSum', name=None):
|
||||
super(ReduceSum, self).__init__(params=['z', 'axis'], op_name=op_name, name=name)
|
||||
self.z = z
|
||||
self.axis = axis
|
||||
self.params_with_defaults = ['axis']
|
||||
|
@ -674,13 +673,6 @@ class Sin(ComputationNode):
|
|||
self.params_with_defaults = []
|
||||
self.inputs = ['z']
|
||||
|
||||
class Softmax(ComputationNode):
|
||||
def __init__(self, z, op_name='Softmax', name=None):
|
||||
super(Softmax, self).__init__(params=['z'], op_name=op_name, name=name)
|
||||
self.z = z
|
||||
self.params_with_defaults = []
|
||||
self.inputs = ['z']
|
||||
|
||||
class Hardmax(ComputationNode):
|
||||
def __init__(self, z, op_name='Hardmax', name=None):
|
||||
super(Hardmax, self).__init__(params=['z'], op_name=op_name, name=name)
|
||||
|
|
|
@ -8,6 +8,17 @@
|
|||
|
||||
from cntk.graph import ComputationNode, _InputComputationNodeBase, _ImageInputComputationNodeBase
|
||||
|
||||
class Slice(ComputationNode):
|
||||
def __init__(self, _, beginIndex, endIndex, axis=1, op_name='CNTK2.Slice',
|
||||
name=None):
|
||||
super(Slice, self).__init__(params=['_', 'beginIndex', 'endIndex', 'axis'], op_name=op_name, name=name)
|
||||
self._ = _
|
||||
self.beginIndex = beginIndex
|
||||
self.endIndex = endIndex
|
||||
self.axis = axis
|
||||
self.inputs = ['_']
|
||||
self.params_with_defaults = ['axis']
|
||||
|
||||
class Ceil(ComputationNode):
|
||||
def __init__(self, _, op_name='CNTK2.Ceil', name=None):
|
||||
super(Ceil, self).__init__(params=['_'], op_name=op_name, name=name)
|
||||
|
@ -37,17 +48,6 @@ class DynamicAxis(ComputationNode):
|
|||
self.params_with_defaults = []
|
||||
self.inputs = []
|
||||
|
||||
class Slice(ComputationNode):
|
||||
def __init__(self, _, beginIndex, endIndex, axis=1, op_name='CNTK2.Slice',
|
||||
name=None):
|
||||
super(Slice, self).__init__(params=['_', 'beginIndex', 'endIndex', 'axis'], op_name=op_name, name=name)
|
||||
self._ = _
|
||||
self.beginIndex = beginIndex
|
||||
self.endIndex = endIndex
|
||||
self.axis = axis
|
||||
self.inputs = ['_']
|
||||
self.params_with_defaults = ['axis']
|
||||
|
||||
class Input(_InputComputationNodeBase):
|
||||
def __init__(self, shape, dynamicAxis='', tag='feature', op_name='CNTK2.Input', name=None):
|
||||
super(Input, self).__init__(params=['shape', 'dynamicAxis', 'tag'], op_name=op_name, name=name)
|
||||
|
|
Загрузка…
Ссылка в новой задаче