Added modified loss views to the loss_calculator.py

As mentioned here, the loss_calculator.py needs to adjust to use modified version of the views designed for losses.

approve @ds @shinde @laurens