Solved by ptrblck in post #2 
              
              
                It’s a bit hard to certainly explain what this code is doing, as all objects are undefined. 
However, based on the used operations my guess is that mask_np would contain class indices, while mask_one_hot represents the one-hot encoded mask as seen here: 
# setup
classes_count = 10
nb_samples = 3
# t…