Search results for pytorch implementation loss functions