Module io 

Functions

load_image

Loads an image from the specified file path. The image is automatically converted to RGB/Grayscale and represented as a [C, H, W] float tensor.

save_image

Saves the image tensor to the specified file path. Assumes image tensor layout is [C, H, W] with float values in [0.0, 1.0].