sde_autoencoder.py 文件源码

def build_encoder(input_var=None, encoder_units=None):
    layer_input = las.layers.InputLayer(shape=(None, 1200),
                                        input_var=input_var)
    layer_encoder = las.layers.DenseLayer(
        layer_input, num_units=encoder_units,
        nonlinearity=las.nonlinearities.sigmoid,
        W=las.init.GlorotUniform())
    layer_decoder = las.layers.DenseLayer(layer_encoder, num_units=1200,
                                          nonlinearity=None,
                                          W=layer_encoder.W.T)
    return layer_decoder