rstsr-core 0.7.3

An n-Dimension Rust Tensor Toolkit
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

use crate::prelude_dev::*;

impl<T, D> DeviceIndexSelectAPI<T, D> for DeviceRayonAutoImpl
where
    T: Clone + Send + Sync,
    D: DimAPI + DimSmallerOneAPI,
    D::SmallerOne: DimAPI,
{
    fn index_select(
        &self,
        c: &mut Vec<MaybeUninit<T>>,
        lc: &Layout<D>,
        a: &Vec<T>,
        la: &Layout<D>,
        axis: usize,
        indices: &[usize],
    ) -> Result<()> {
        let pool = self.get_current_pool();
        index_select_cpu_rayon(c, lc, a, la, axis, indices, pool)
    }
}