Pytorch Custom Loss Function Tensorflow Python