Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- class Decoder(tf.keras.layers.Layer):
- def __init__(self, intermediate_dim, original_dim):
- super(Decoder, self).__init__()
- self.hidden_layer = tf.keras.layers.Dense(units=intermediate_dim, activation=tf.nn.relu)
- self.output_layer = tf.keras.layers.Dense(units=original_dim, activation=tf.nn.relu)
- def call(self, code):
- activation = self.hidden_layer(code)
- return self.output_layer(activation)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement