Enable wrap_dim in Local.cwrap.

This includes torch.cat, which is a TensorList argument, which wasn't supported before.
5 files changed