test_layer_normalization_RNN.py 文件源码

def BuildModel(dim):
    '''
    :param dim: input shape.

    The Separable_SimpleRNN can be added after linear convolution.
    Make sure the nb_filter in the Convolution layer is equivalent to output_dim in RNN
    '''
    inp = Input(shape=dim)
    #
    x_conv = Convolution1D(nb_filter=16, filter_length=5, border_mode='same')(inp)
    x_rnn = LN_SimpleRNN(output_dim=16, activation='tanh')(x_conv)
    #
    model = Model(input=inp, output=x_rnn)
    return model


# Load an example image.