Is your feature request related to a problem?
Currently, the Image Class uses pillow.
Desired solution
The Image Class should use PyTorch tensors and methods from the PyTorch functional API (v2)
Possible alternatives (optional)
No response
Screenshots (optional)
No response
Additional Context (optional)
No response