Exemplos de broadcast_and_squeeze_mixed em Python

Exemplo n.º 1

Arquivo: distributions.py Projeto: wangyongguang/Brancher

 def _preprocess_parameters_for_log_prob(self, x, **parameters):
     tuple_x, parameters = broadcast_and_squeeze_mixed(
         tuple([x]), parameters)
     return tuple_x[
         0], parameters, None, None  #TODO: add proper output here

Exemplo n.º 2

Arquivo: distributions.py Projeto: wangyongguang/Brancher

 def _preprocess_parameters_for_sampling(self, **parameters):
     parameters = broadcast_and_squeeze_mixed((), parameters)
     return parameters, None