[keras] layers: adds misc layers

keras.py

@@ -1082,6 +1082,26 @@ class SplitHighLow(tf.keras.layers.Layer):
         return inputs[:, :, :4], inputs[:, :, 4:]
 
 
+class LL(tf.keras.layers.Layer):
+    def call(self, inputs):
+        return inputs[:, :, :4]
+
+
+class OneHotPDGID(tf.keras.layers.Layer):
+    def call(self, inputs):
+        return tf.one_hot(tf.cast(tf.math.abs(inputs[:, :, -2]) > 11, tf.int32), 2)
+
+
+class OneHotCharge(tf.keras.layers.Layer):
+    def call(self, inputs):
+        return tf.one_hot(tf.cast(inputs[:, :, -1] > 0, tf.int32), 2)
+
+
+class BlackBtag(tf.keras.layers.Layer):
+    def call(self, inputs):
+        return tf.keras.layers.Concatenate()([inputs[:, :, :-1], 0 * inputs[:, :, -1:]])
+
+
 class LBNLayer(tf.keras.layers.Layer):
     """
     Custom implementation of the LBNLayer with automatic cropping to