TensorFlow 2.0 Beta is available Learn more

tf.contrib.model_pruning.get_masks

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