if (t==0) argmax(logits) else pick(logits)