Function tch::nn::func

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