Add impl for torchvision.nms and support multiple batch dim for convolution.

Add impl for torchvision.nms and support multiple batch dim for convolution. #873