I use a wrench but only on difficult bends. You have to be careful you don't do more damage on a hot shaft by denting it with the wrench.
Once heated cane, especially, plasticizes well so you can hand straighten most bends by hand. The secret is to not overwork the heated shaft. Straighten a bend and set it aside to cool before moving to the next bend.