Hi there, I am trying to get the simple PixelCNN model working on CIFAR10, I took a look at your implementation here https://github.com/pbloem/pixel-models/blob/master/layers.py#L8-L32, and it looks like the masks are only on pixels, but there are no masks among RGB values, is this correct?