pub fn load<T: AsRef<Path>>(path: T) -> Result<Tensor, TchError>
Loads an image from a file.
On success returns a tensor of shape [channel, height, width].