understanding == applied to a numpy array,np.array([(labels == i).astype(np.float32) for i in np.arange(3)])