Skip to main content

apply_conditional_elementwise

torsh_functional::utils

Function apply_conditional_elementwise 

Source 
pub fn apply_conditional_elementwise<F>(
    input: &Tensor,
    condition: F,
    true_op: impl Fn(f32) -> f32,
    false_op: impl Fn(f32) -> f32,
    _inplace: bool,
    _context: &str,
) -> Result<Tensor>
where
    F: Fn(f32) -> bool,
Expand description

Utility for conditional element-wise operations (like ReLU, LeakyReLU, etc.)