W3cubDocs

/TensorFlow Python

tf.contrib.model_pruning.apply_mask

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

Defined in tensorflow/contrib/model_pruning/python/pruning.py.

Apply mask to a given weight tensor.

Args:

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

Returns:

Tensor representing masked_weights

© 2018 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/contrib/model_pruning/apply_mask