Function tch::nn::func

pub fn func<'a, F>(f: F) -> Func<'a>where
    F: 'a + Fn(&Tensor) -> Tensor + Send,