Exemplo n.º 1
0
def test_rnn_block():
    utils.block_basic_exam(
        basic.RNNBlock(),
        tf.keras.Input(shape=(32, 10), dtype=tf.float32),
        [
            'bidirectional',
            'layer_type',
            'num_layers',
        ])
Exemplo n.º 2
0
 def build(self, hp, inputs=None):
     input_node = nest.flatten(inputs)[0]
     output_node = input_node
     output_node = basic.RNNBlock().build(hp, output_node)
     return output_node