torch_choice
