Esempi in Python per unit_targeting

Esempio n. 1

File: pruning_utils.py Progetto: 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

Esempio n. 2

File: pruning_utils.py Progetto: 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