My PC info: OS: Ubuntu 18.04, CPU: i7-9700K, Wrong ouput
Laptop info: OS: Windows 10, CPU: i7-8565U, Wrong ouput
My GPU Server: OS: CentOS 7.7, CPU: Intel® Xeon® Gold 6230, Correct output
PyTorch tested: 1.3.1, 1.4, 1.5. Python tested: 3.6, 3.7