tf.contrib.model_pruning.get_masked_weights

View source on GitHub 
tf.contrib.model_pruning.get_masked_weights()