Solved by KFrank in post #2 
              
              
                Hi Sergey! 

Support for “soft,” probabilistic targets for CrossEntropyLoss is new 
as of (I believe) the current stable version, 1.10.0. 
Your best bet will be to upgrade to the current stable release. 
Best. 
K. Frank