Solved by KFrank in post #6 
              
              
                Hello Blade! 

I have two networks that are trained together. 
… 
What you see is my downstream network: it takes in output of the upstream W 
… 
I want to turn them into either a one-hot vector or class labels W2 


Once you turn the output of your upstream network into a one-hot 
vector or a class…