hodu_nn 0.1.0

hodu nn
Documentation 
1
2
3
4
5
6
7

use hodu_core::{error::HoduResult, tensor::Tensor};
pub use hodu_nn_macros::Module;

pub trait Module<I = &'static Tensor> {
    fn forward(&self, input: I) -> HoduResult<Tensor>;
    fn parameters(&mut self) -> Vec<&mut Tensor>;
}