def create_one_hot_sketch(self, element, all_sketches, cm_width):
    """Wrapper to create the initial sketch for an one-hot vector."""
    element = tf.expand_dims(element, axis=1)
    softmax = tf.ones_like(element, dtype=tf.float32)
    return module.create_cm_sketch(element, softmax, all_sketches, cm_width)
Wrapper to create the initial sketch for an one-hot vector.
language/emql/model.py