model.py 文件源码

def decode(self, x, test=False, sigmoid=True):

        activate = F.relu

        # Hidden
        h = x
        for i in range(self.n_layers_dec - 1):
            h = getattr(self, "dec_layer_%i" % i)(h)
            h = getattr(self, "dec_batchnorm_%i" % i)(h, test=test)

            h = activate(h)
            if self.dropout:
                h = F.dropout(h, train=not test)

        # Output
        output = getattr(self, "dec_layer_out")(h)
        if sigmoid:
            output = F.sigmoid(output)

        return output