Weird error about "packing tensors" when sampling with enumeration

Awesome, thank you very much! :smile: