Hi @arora_aman,
I implemented a rgb_transform on GPU, you can find the code here
What this transform does is to change only one channel values of the whole batch.
Is didn’t suggest a PR of it because I would like to implemented another version that only affects part of the batch (or at least at CPU level) but I am kind of busy these days.