Custom Loss Example - Bigdl Typing CST Test