Searched defs:gumbel_softmax (Results 1 – 2 of 2) sorted by relevance
186 inline Tensor gumbel_softmax( in gumbel_softmax() function
2146 def gumbel_softmax( function