Skip to main content

ruby_optimizer_step

Function ruby_optimizer_step 

Source
#[unsafe(no_mangle)]
pub unsafe extern "C" fn ruby_optimizer_step( optimizer: *mut TorshOptimizer, _parameters: *mut *mut TorshTensor, _gradients: *mut *mut TorshTensor, _param_count: usize, ) -> TorshError
Expand description

Ruby-specific wrapper for optimizer step