Identical Transformation on Input and Label?

Hi

I had similar problem and I solved it. Here is the link to the post. Although, I have used a subset of original transformation pipeline, still I wanted to have identical for that subset.

bests,
Nik