Complex value support?

Hi,

Thanks for putting this together - complex number support is starting to get pretty advanced in pytorch, is there any chance you'd consider changing things here to allow for it?

Cheers,

Stephen