torch_sys::c_generated

Function atg_feature_dropout

source
pub unsafe extern "C" fn atg_feature_dropout(
    out__: *mut *mut C_tensor,
    input_: *mut C_tensor,
    p_: f64,
    train_: c_int,
)