Searched defs:head_logprob (Results 1 – 1 of 1) sorted by relevance
159 const Tensor head_logprob = F::log_softmax(head_output, 1); in forward() local174 const Tensor head_logprob = F::log_softmax(head_output, 1); in _get_full_log_prob() local