Pass datapoint wise mask to prox / LMO for accelerating line search

[GeoffNN]

We should be able to compute operators such as prox and LMO for only a subset of the batch.

For example, when performing back-tracking line-search, we only want to update certain datapoints in the batch within the backtracking loop. For now the computation is redundantly done for all datapoints, then masked.