用PyTorch训练一个简单的K-means无监督算法机器学习模型