class_docstr: A custom layer which is not prunable.
class_docstr_tok: ["A","custom","layer","which","is","not","prunable","."]
code: def __init__(self, input_dim=16, output_dim=32):
      super(CustomLayer, self).__init__()
      self.weight = self.add_weight(
          shape=(input_dim, output_dim),
          initializer="random_normal",
          trainable=True,
      )
      self.bias = self.add_weight(
          shape=(output_dim), initializer="zeros", trainable=True
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_schedule
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_wrapper
label_desc: Use image node_id 1 to create a new CustomLayer object from inherited base classes: keras.layers with example: obj = CustomLayer(input_dim, output_dim)