def __init__(self, **kwargs):
     kwargs.pop(str('config'))
     Lambda.__init__(self, **kwargs)
Exemple #2
0
 def __init__(self, name=None):
     Lambda.__init__(self, self.fn, name=name)
Exemple #3
0
 def __init__(self, axis=-1):
     Lambda.__init__(self, lambda x: tf.squeeze(x, axis=axis))