* Fix indexing error in nms kernel

Here it is indexing a cuda tensor with CPU indices. It used to work, but 
after https://github.com/pytorch/pytorch/commit/006505bb8f9dcf0f38b32518308d071c8a1ccec6 it results in memory corruption.

* Use the device of other tensors