pytorch常用函数手册
时间: 2023-11-18 15:04:02 浏览: 587
PyTorch是一个开源的机器学习框架,提供了丰富的函数库来支持深度学习任务。以下是一些常用的PyTorch函数:
1. torch.Tensor:创建张量。
2. torch.autograd:自动求导。
3. torch.nn:神经网络模块和损失函数。
4. torch.optim:优化器。
5. torch.utils.data:数据加载和预处理。
6. torchvision:计算机视觉库。
7. torch.functional:包含一些不需要状态的函数,如ReLU、Sigmoid、Softmax等。
8. torch.sparse:稀疏张量。
9. torch.distributions:概率分布函数。
10. torch.utils:包含一些实用工具,如数据集、学习率调度器等。
阅读全文
相关推荐












