native_neural_network 0.1.6

Lib no_std Rust for native neural network (.rnn)
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
26
27
28
29

pub fn conv5d_output_dim(input: usize, kernel: usize, pad: usize, stride: usize) -> Option<usize> {
    if stride == 0 || kernel == 0 {
        return None;
    }
    let padded = input.checked_add(pad.checked_mul(2)?)?;
    if padded < kernel {
        return Some(0);
    }
    Some((padded - kernel) / stride + 1)
}

pub struct Conv5dOutputShapeArgs {
    pub d: usize,
    pub h: usize,
    pub w: usize,
    pub kd: usize,
    pub kh: usize,
    pub kw: usize,
    pub pad: (usize, usize, usize),
    pub stride: (usize, usize, usize),
}

pub fn conv5d_output_shape(args: Conv5dOutputShapeArgs) -> Option<(usize, usize, usize)> {
    Some((
        conv5d_output_dim(args.d, args.kd, args.pad.0, args.stride.0)?,
        conv5d_output_dim(args.h, args.kh, args.pad.1, args.stride.1)?,
        conv5d_output_dim(args.w, args.kw, args.pad.2, args.stride.2)?,
    ))
}