Mean Average Precision Pytorch Cuda