Function tch::nn::func_t

source ·
pub fn func_t<'a, F>(f: F) -> FuncT<'a>where
    F: 'a + Fn(&Tensor, bool) -> Tensor + Send,