Currently C(constraint, X) only returns the contracted box.
But it calculates C.forward(X), i.e. the forward image under the function, as the first part of the process.

Given that it calculates this and this info is useful, it should return it.