Keras one-hot-encode layer

keras_one_hot_encode.py

class OneHotEncode(Embedding):
    """
    One-hot-encode layer eg. [[1, 3]] -> [[[0,1,0,0],[0,0,0,1]]]
    
    input_dim: Number of categories to one-hot encode
    input_length: length of X vecotrs (optional)
    mask_zero: treat 0 as masking
    """
    
    def __init__(self, input_dim, input_length=None, mask_zero=False):
        super().__init__(input_dim=input_dim, output_dim=input_dim, input_length=input_length,
                         embeddings_initializer='identity', trainable=False, mask_zero=mask_zero)