optirs-core 0.3.2

OptiRS core optimization algorithms and utilities
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
30
31
32

use optirs_core::regularizers::SpatialDropout;
use scirs2_core::ndarray::{Array, Axis};

#[allow(dead_code)]
fn main() {
    let sd = SpatialDropout::new(0.0).expect("unwrap failed"); // No dropout to test

    // Create a simple 4D tensor (batch, channels, height, width)
    let features = Array::from_shape_fn((1, 2, 2, 2), |(b, c, h, w)| {
        println!("Creating element at ({}, {}, {}, {})", b, c, h, w);
        c as f64 + h as f64 + w as f64
    });

    println!("Original features shape: {:?}", features.shape());
    println!("Original features:\n{:?}", features);

    // Apply with no dropout
    let result = sd.apply(&features, true);
    println!("Result shape: {:?}", result.shape());
    println!("Result:\n{:?}", result);

    // Test with dropout = 0.5
    let sd_drop = SpatialDropout::new(0.5).expect("unwrap failed");
    let result_drop = sd_drop.apply(&features, true);
    println!("Result with dropout:\n{:?}", result_drop);

    // Check channel axis
    for c in 0..2 {
        let channel = result_drop.index_axis(Axis(1), c);
        println!("Channel {}: {:?}", c, channel);
    }
}