Ejemplos de unit_targeting en Python

Ejemplo n.º 1

Archivo: pruning_utils.py Proyecto: zxChouSean/tensor2tensor

def unit(w, sparsity):
  """Unit-level magnitude pruning."""
  w_shape = common_layers.shape_list(w)
  count = tf.to_int32(w_shape[-1] * sparsity)
  mask = common_layers.unit_targeting(w, count)
  return (1 - mask) * w

Ejemplo n.º 2

Archivo: pruning_utils.py Proyecto: qixiuai/tensor2tensor

def unit(w, sparsity):
  """Unit-level magnitude pruning."""
  w_shape = common_layers.shape_list(w)
  count = tf.to_int32(w_shape[-1] * sparsity)
  mask = common_layers.unit_targeting(w, count)
  return (1 - mask) * w