import tensorflow as tf


    # Create a tensot flow constant equal to the number of classes
    C = tf.constant(N_classes, name="C")
    one_hot_matrix = tf.one_hot(vect-1, C, axis=0) #axis=0 means it is mapping to column vectors
