singe-kernel 0.1.0-alpha.4

Reusable CPU and GPU kernels.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

//! Copies, transposes, padding, slicing, tiling, and layout transforms.

pub fn transpose_2d(input: &[f32], rows: usize, cols: usize) -> Vec<f32> {
    let mut output = vec![0.0; input.len()];
    for row in 0..rows {
        for col in 0..cols {
            output[col * rows + row] = input[row * cols + col];
        }
    }
    output
}

pub fn pack_downsample_2d(input: &[f32], rows: usize, cols: usize, factor: usize) -> Vec<f32> {
    let mut output = vec![0.0; input.len()];
    let output_rows = rows / factor;
    let output_cols = cols * factor;
    for row in 0..output_rows {
        for packed in 0..output_cols {
            let offset = packed / cols;
            let feature = packed % cols;
            output[row * output_cols + packed] = input[(row * factor + offset) * cols + feature];
        }
    }
    output
}