register_nn_module

torsh_python::nn

Function register_nn_module 

Source 
pub fn register_nn_module(
    _py: Python<'_>,
    m: &Bound<'_, PyModule>,
) -> PyResult<()>
Expand description

Register the nn module with Python