How to apply a function for each image in the batch?

vision
1

Hello.
My goal is to apply a user-defined function for each image in the batch, in the forward propagation.
For example, batch size= 12 and I want to apply py360convert.e2c() function on each image in the batch.

In pandas, df.apply() does this job, however I could not find an example for PyTorch.
Is there any approach?

Thanks in advance.