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 |