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