Searched defs:cluster_logprob (Results 1 – 1 of 1) sorted by relevance
137 const Tensor cluster_logprob = F::log_softmax(cluster_output, 1); in forward() local184 const Tensor cluster_logprob = F::log_softmax(cluster_output, 1); in _get_full_log_prob() local