pytorch

PyTorch 是一个 Python 深度学习框架，提供带 GPU 加速的张量计算与基于动态图的自动求导，是科研与生产中训练神经网络的主流选择之一。