Esempio n. 1
0
 def __init__(self, axis=-1):
     Lambda.__init__(self, lambda x: tf.squeeze(x, axis=axis))
Esempio n. 2
0
 def __init__(self, size):
     Lambda.__init__(self, lambda x: tf.tile(tf.expand_dims(tf.expand_dims(x, 2), 3), [1, 1, size, size]))
Esempio n. 3
0
 def __init__(self, scale):
     Lambda.__init__(self, lambda x: tf.cast(x, tf.float32) * scale)
Esempio n. 4
0
 def __init__(self):
     Lambda.__init__(self, lambda x: tf.log(x + 1e-10))
Esempio n. 5
0
 def __init__(self, dims):
     Lambda.__init__(self, lambda x: tf.transpose(x, dims))
Esempio n. 6
0
 def __init__(self, num_splits=2, axis=-1):
     Lambda.__init__(self, lambda x: tf.split(x, num_splits, axis=axis))