code: def per_batch_transform_on_device(self):
      return ApplyToKeys(
          DataKeys.INPUT,
          Ka.RandomHorizontalFlip(p=0.25),
      )
  
file_path: lightning-flash/examples/image/learn2learn_img_classification_imagenette.py
imports: import warnings
  from dataclasses import dataclass
  from typing import Tuple, Union
  import flash
  import kornia.augmentation
  import kornia.geometry
  import learn2learn
  import torch
  import torchvision.transforms
  from flash.core.data.io.input import DataKeys
  from flash.core.data.io.input_transform import InputTransform
  from flash.core.data.transforms import ApplyToKeys
  from flash.image import ImageClassificationData, ImageClassifier
label_desc: Use image node_id 3 for calling the ImageClassificationInputTransform obj's underlying member method code with example usage: obj.per_batch_transform_on_device() and returns: ApplyToKeys