gru_layers.py 文件源码

def param_init_decoder(options, params, prefix='decoder_gru'):

    n_x = options['n_x']
    n_h = options['n_h']

    W = np.concatenate([uniform_weight(n_x,n_h),
                        uniform_weight(n_x,n_h)], axis=1)
    params[_p(prefix,'W')] = W

    U = np.concatenate([ortho_weight(n_h),
                        ortho_weight(n_h)], axis=1)
    params[_p(prefix,'U')] = U

    params[_p(prefix,'b')] = zero_bias(2*n_h)

    Wx = uniform_weight(n_x, n_h)
    params[_p(prefix,'Wx')] = Wx

    Ux = ortho_weight(n_h)
    params[_p(prefix,'Ux')] = Ux

    params[_p(prefix,'bx')] = zero_bias(n_h)

    params[_p(prefix,'b0')] = zero_bias(n_h)

    return params