Custom Loss Example - Pytorch Typing CST Test