Pytorch Custom Loss Function In Xgboost