adding prox for all dim, addding test prox with 2d, 3d, 4d matrices