In fact not even cpu and gpu versions match
Tensor type:float32
Numpy array type: float32
Numpy array type: torch.float32
CPU-CUDA: 0.5086007714271545
CPU-NP: 0.3293250799179077
CUDA-NP: 0.37688499689102173