核心组件 | 描述 |
---|---|
torch | 类似NumPy的张量库,拥有强大GPU支持 |
torch.autograd | 基于磁带的自动微分库,支持torch中所有可微分的张量操作 |
torch.jit | 编译堆栈(TorchScript),用于从PyTorch代码创建可序列化和可优化的模型 |
torch.nn | 与自动梯度深度集成的神经网络库,追求最大限度的灵活性 |
torch.multiprocessing | Python多进程处理,但在进程之间以神奇的方式共享torch张量内存。适用于数据加载和Hogwild训练 |
torch.utils | 数据加载器和其他易用的实用函数 |
原文地址:https://blog.csdn.net/weixin_45725295/article/details/134719746
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如若转载,请注明出处:http://www.7code.cn/show_17963.html
如若内容造成侵权/违法违规/事实不符,请联系代码007邮箱:suwngjj01@126.com进行投诉反馈,一经查实,立即删除!