Won't this cause the loss_mask to be misaligned?
Won't this cause the loss_mask to be misaligned?