af461df627ef660d1a71b6981dedb4e4b504ba9a,txtgen/modules/connectors/connectors.py,StochasticConnector,_build,#StochasticConnector#Any#,256

Before Change

        sampler = get_function(self.hparams.distribution, modules)

        if sampler is distributions.sample_gaussian:
            if not isinstance(inputs, tuple):
                raise ValueError(
                    "Gaussian connector requires tuple input tensors.")

            if len(inputs) == 2:
                input_mu, input_log_var = inputs
                output = sampler(input_mu, input_log_var)

After Change

        output = inputs.sample()

        try:
            nest.assert_same_structure(inputs, self._decoder_state_size)
        except (ValueError, TypeError):
            flat_input = nest.flatten(inputs)
            output = nest.pack_sequence_as(
                self._decoder_state_size, flat_input)

Instances