Skip to main content

ruby_linear_forward

torsh_ffi::ruby

Function ruby_linear_forward 

Source 
#[unsafe(no_mangle)]
pub unsafe extern "C" fn ruby_linear_forward(
    module: *mut TorshModule,
    input: *mut TorshTensor,
) -> *mut TorshTensor
Expand description

Ruby-specific wrapper for linear layer forward pass