Function create_mlp 

pub fn create_mlp(
    name: &str,
    input_size: usize,
    _hidden_size: usize,
    _output_size: usize,
    weights: Vec<&Array>,
) -> Result<OnnxModel>

Create a simple feedforward neural network model

Arguments

• name - Model name
• input_size - Input feature size
• hidden_size - Hidden layer size
• output_size - Output size
• weights - Layer weights [w1, b1, w2, b2]

Example

use numrs::ops::model::create_mlp;
use numrs::Array;

let w1 = Array::new(vec![784, 128], vec![0.0; 784 * 128]);
let b1 = Array::new(vec![128], vec![0.0; 128]);
let w2 = Array::new(vec![128, 10], vec![0.0; 128 * 10]);
let b2 = Array::new(vec![10], vec![0.0; 10]);

let model = create_mlp("mnist_classifier", 784, 128, 10,
                       vec![&w1, &b1, &w2, &b2])?;