示例#1
0
 def _init_distribution(self):
     if self.pooling:
         DiagGauss.__init__(self, self.rnn.output[:, : self.n_output], self.rnn.output[:, self.n_output :], self.rng)
     else:
         DiagGauss.__init__(
             self, self.rnn.output[:, :, : self.n_output], self.rnn.output[:, :, self.n_output :], self.rng
         )
示例#2
0
 def _init_distribution(self):
     if self.pooling:
         DiagGauss.__init__(self, self.rnn.output[:, :self.n_output],
                            self.rnn.output[:, self.n_output:], self.rng)
     else:
         DiagGauss.__init__(self, self.rnn.output[:, :, :self.n_output],
                            self.rnn.output[:, :, self.n_output:], self.rng)
示例#3
0
 def _init_distribution(self):
     var = self.std**2 + 1e-5
     if self.pooling:
         DiagGauss.__init__(self, self.rnn.output[:, :self.n_output],
                            var.dimshuffle('x', 0), self.rng)
     else:
         DiagGauss.__init__(self, self.rnn.output[:, :, :self.n_output],
                            var.dimshuffle('x', 'x', 0), self.rng)
示例#4
0
 def _init_distribution(self):
     var = self.std ** 2 + 1e-5
     if self.pooling:
         DiagGauss.__init__(
             self,
             self.rnn.output[:, :self.n_output],
             var.dimshuffle('x', 0),
             self.rng)
     else:
         DiagGauss.__init__(
             self,
             self.rnn.output[:, :, :self.n_output],
             var.dimshuffle('x', 'x', 0),
             self.rng)