pub fn load_from_memory(img_data: &[u8]) -> Result<Tensor, TchError>
Loads an image from memory.
On success returns a tensor of shape [channel, height, width].
