PyTorch

用张量运算、自动求导和神经网络模块构建、训练与运行深度学习模型。