pub fn load_tensor<B: Backend, const D: usize>(
    name: &str,
    path: &str,
    device: &B::Device
) -> Result<Tensor<B, D>, Box<dyn Error>>