Skip to main content

dropout

yscv_kernels

Function dropout 

Source 
pub fn dropout(
    input: &Tensor,
    p: f32,
    seed: u64,
    training: bool,
) -> Result<Tensor, KernelError>
Expand description

Applies dropout: randomly zeroes elements with probability p.

During inference (training=false), returns input unchanged. Uses xorshift64 PRNG with given seed for deterministic masking.