Add `stack` for concatenating along new axis

[joelberkeley]

It's a nuisance to have to add a new dim before concatenating tensors. Add stack which does both in one.