pytorch backend
2017-04-03 10:59:26 0 举报
登录查看完整内容
PyTorch后端(backend)是PyTorch框架中用于执行计算操作的底层库。它负责处理张量运算、自动微分、优化器等功能,并将这些功能与硬件设备(如CPU、GPU等)进行高效交互。PyTorch支持多种后端,如TensorFlow后端(TorchScript)、Microsoft Cognitive Toolkit(CNTK)后端等,用户可以根据需要选择合适的后端以获得最佳性能。此外,PyTorch还提供了灵活的API,使得用户可以方便地在不同后端之间切换。总之,PyTorch后端为深度学习模型的训练和推理提供了强大的计算支持。
作者其他创作
大纲/内容
ATen
backend C++ API.cpp.h
C++ binding
Python API
configs
TH
Python interpreter
Compiler / Interpreter
CPU
support params
PyTorch
high-level interface
Hardware
THC
backend C++ API.cu.h
GPU
gcc x64
0 条评论
回复 删除
下一页