Function load

pub fn load<P: AsRef<Path>>(
    filename: P,
    device: &Device,
) -> Result<HashMap<String, Tensor>>