Ejemplos de get_masked_array en Python

Ejemplo n.º 1

Archivo: empirical.py Proyecto: yorkerlin/neural-tangents

 def output(x, **kwargs):
     out = f(params, x, **kwargs)
     masked_output = utils.get_masked_array(out)
     return utils.nt_tree_fn()(lambda x: x.masked_value)(masked_output)

Ejemplo n.º 2

 def output(x, **kwargs):
   out = f(params, x, **kwargs)
   masked_output = utils.get_masked_array(out)
   return masked_output.masked_value

Ejemplo n.º 3

 def _f(p):
   out = f(p, x, **apply_fn_kwargs)
   # TODO(romann): normalize properly if output is masked.
   out = utils.get_masked_array(out)
   return out.masked_value

Ejemplo n.º 4

 def output(x, rng):
   out = f(params, x, rng=rng, **apply_fn_kwargs)
   masked_output = utils.get_masked_array(out)
   return masked_output.masked_value