promptbind.utils.utils.gumbel_softmax_no_random

promptbind.utils.utils.gumbel_softmax_no_random(logits: Tensor, tau: float = 1, hard: bool = False, eps: float = 1e-10, dim: int = -1) Tensor[source]