Пример #1
0
 def __init__(self, axis=-1):
     Lambda.__init__(self, lambda x: tf.squeeze(x, axis=axis))
Пример #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]))
Пример #3
0
 def __init__(self, scale):
     Lambda.__init__(self, lambda x: tf.cast(x, tf.float32) * scale)
Пример #4
0
 def __init__(self):
     Lambda.__init__(self, lambda x: tf.log(x + 1e-10))
Пример #5
0
 def __init__(self, dims):
     Lambda.__init__(self, lambda x: tf.transpose(x, dims))
Пример #6
0
 def __init__(self, num_splits=2, axis=-1):
     Lambda.__init__(self, lambda x: tf.split(x, num_splits, axis=axis))