<s> def test_gru_network ( ) : <EOL> from rllab . core . network import GRUNetwork <EOL> import lasagne . layers as L <EOL> from rllab . misc import ext <EOL> import numpy as np <EOL> network = GRUNetwork ( <EOL> input_shape = ( <NUM_LIT:2> , <NUM_LIT:3> ) , <EOL> output_dim = <NUM_LIT:5> , <EOL> hidden_dim = <NUM_LIT:4> , <EOL> ) <EOL> f_output = ext . compile_function ( <EOL> inputs = [ network . input_layer . input_var ] , <EOL> outputs = L . get_output ( network . output_layer ) <EOL> ) <EOL> assert f_output ( np . zeros ( ( <NUM_LIT:6> , <NUM_LIT:8> , <NUM_LIT:2> , <NUM_LIT:3> ) ) ) . shape == ( <NUM_LIT:6> , <NUM_LIT:8> , <NUM_LIT:5> ) </s>
