Yeah the slicing definitely seems fishy. We will look into that, thanks a lot.
Regarding the softmax, it is part of the WaveNet structure as it passes in the output. The architechture below:

image1013×538 69.6 KB

Where should the softmax then be implemented, if output cannot be passed through said function for our loss?
Appreciate the help!!! (Been stuck here for a week)