Make inc/set_subtensor work on output of take.