Hi berak:
Yes, the code is a product of my experimentation with PyTorch. I am a raw beginner, so I might be doing something counterproductive. I see why you’d use the Softmax, it gives a set of probabilities that sum to 1. Reminds me of information theory. Thanks for the tip! Much appreciated.
- Shawn