pytorch的顶层命名空间
时间: 2023-10-27 15:43:52 浏览: 108
BERT-NER:用BERT的Pytorch命名实体识别
PyTorch的顶层命名空间包括以下模块:
1. torch:包含各种基本操作和数据类型,如张量、数值计算、文件I/O、随机数生成等。
2. torch.autograd:用于自动求导,可以自动计算张量的梯度。
3. torch.nn:包含各种神经网络层、损失函数等。
4. torch.optim:包含各种优化器,如SGD、Adam等。
5. torch.utils:包含各种工具函数,如数据集加载器、学习率调度器等。
6. torch.onnx:用于将PyTorch模型转换为ONNX格式。
7. torch.distributions:包含各种概率分布,如正态分布、伯努利分布等。
8. torch.multiprocessing:用于多进程编程。
9. torch.jit:用于将PyTorch模型编译为高效的本地代码。
10. torch.fft:包含各种快速傅里叶变换相关操作。
阅读全文