TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.contrib.model_pruning.apply_mask

View source on GitHub

Apply mask to a given weight tensor.

tf.contrib.model_pruning.apply_mask(
    x,
    scope=''
)

Args:

  • x: Input weight tensor
  • scope: The current variable scope. Defaults to "".

Returns:

Tensor representing masked_weights