簡體   English   中英

有沒有辦法轉換ChunkMut <T> 從Vec :: chunks_mut到切片&mut [T]?

[英]Is there a way to convert a ChunkMut<T> from Vec::chunks_mut to a slice &mut [T]?

我正在並行填充向量,但是對於這個一般性問題,我只發現了提示而沒有答案。

下面的代碼有效,但是我想切換到Rng::fill而不是遍歷每個塊。 一個Vec內可能沒有多個可變切片; 我不確定。

extern crate rayon;
extern crate rand;
extern crate rand_xoshiro;

use rand::{Rng, SeedableRng};
use rand_xoshiro::Xoshiro256StarStar;
use rayon::prelude::*;
use std::{iter, env};
use std::sync::{Arc, Mutex};

// i16 because I was filling up my RAM for large tests...
fn gen_rand_vec(data: &mut [i16]) {
    let num_threads = rayon::current_num_threads();
    let mut rng = rand::thread_rng();
    let mut prng = Xoshiro256StarStar::from_rng(&mut rng).unwrap();
    // lazy iterator of fast, unique RNGs
    // Arc and Mutex are just so it can be accessed from multiple threads
    let rng_it = Arc::new(Mutex::new(iter::repeat(()).map(|()| {
        let new_prng = prng.clone();
        prng.jump();
        new_prng
    })));
    // generates random numbers for each chunk in parallel
    // par_chunks_mut is parallel version of chunks_mut
    data.par_chunks_mut(data.len() / num_threads).for_each(|chunk| {
        // I used extra braces because it might be required to unlock Mutex. 
        // Not sure.
        let mut prng = { rng_it.lock().unwrap().next().unwrap() };
        for i in chunk.iter_mut() {
            *i = prng.gen_range(-1024, 1024);
        }
    });
}

事實證明, ChunksMut迭代器提供切片。 我不確定如何從文檔中收集信息。 我通過閱讀找出來了:

#[derive(Debug)]
#[stable(feature = "rust1", since = "1.0.0")]
pub struct ChunksMut<'a, T:'a> {
    v: &'a mut [T],
    chunk_size: usize
}

#[stable(feature = "rust1", since = "1.0.0")]
impl<'a, T> Iterator for ChunksMut<'a, T> {
    type Item = &'a mut [T];

    #[inline]
    fn next(&mut self) -> Option<&'a mut [T]> {
        if self.v.is_empty() {
            None
        } else {
            let sz = cmp::min(self.v.len(), self.chunk_size);
            let tmp = mem::replace(&mut self.v, &mut []);
            let (head, tail) = tmp.split_at_mut(sz);
            self.v = tail;
            Some(head)
        }
}

我想這只是經驗。 對於其他人來說,很明顯,來自Vec<T> ChunksMut<T>類型的迭代器會返回[T]類型的對象。 現在這很有意義。 對於中間結構,還不是很清楚。

pub fn chunks_mut(&mut self, chunk_size: usize) -> ChunksMut<T>
// ...
impl<'a, T> Iterator for ChunksMut<'a, T>

讀取此內容后,看起來迭代器返回的對象類型為T ,與Vec<T>.iter() ,這沒有任何意義。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM