Soft Target Classification Loss - Coding problem | OneNoughtOne