Function mt

cake-core/src/models/vibevoice/vae_decoder.rs:533–539

(shape: &[usize], seed: u64)

        let dev = Device::Cpu;

        fn mt(shape: &[usize], seed: u64) -> Tensor {
            use rand::{Rng, SeedableRng};
            let n: usize = shape.iter().product();
            let mut rng = rand::rngs::StdRng::seed_from_u64(seed);
            let d: Vec<f32> = (0..n).map(|_| rng.gen_range(-0.01..0.01)).collect();
            Tensor::from_vec(d, shape, &Device::Cpu).unwrap()
        }

        map.insert("norm.weight".into(), Tensor::ones(ch, DType::F32, &dev).unwrap());
        map.insert("gamma".into(), Tensor::ones(ch, DType::F32, &dev).unwrap());