可以写一个pytorch版本的么

当然可以，PyTorch是一个非常流行的深度学习框架，它提供了丰富的工具和库，可以帮助开发者快速构建和训练深度神经网络模型。如果您需要使用PyTorch来开发深度学习模型，可以参考PyTorch官方文档和示例代码，或者参考其他开源项目的实现。