Function tch::with_grad

pub fn with_grad<T, F>(f: F) -> T where
    F: FnOnce() -> T, 

Runs a closure explicitly keeping track of gradients, this could be run within a no_grad closure for example.