Function tch::autocast

pub fn autocast<T, F>(enabled: bool, f: F) -> T
where
    F: FnOnce() -> T,

Runs a closure in mixed precision.