torch.multinomial with arbitrary number of dimensions, and number of candidates on the last dimension. Args: input (torch.Tensor): The input tensor containing probabilities. num_samples (int): Number of samples to draw. replacement (bool): Whether to draw with replacemen
(input: torch.Tensor, num_samples: int, replacement=False, *, generator=None)
source not stored for this graph (policy: none)
no outgoing calls
no test coverage detected