Assuming the difference between both installations would be a CPU-only vs. CUDA10.2 install, I cannot reproduce the issue and both pip wheels give tensor([[1, 2, 3]]) as the output using your code snippet.