PyTorch包含那些函数
时间: 2024-05-22 16:15:55 浏览: 67
pytorch常用函数手册
PyTorch包含了很多函数,以下是一些常见的函数:
1. torch.Tensor:创建张量的主要类。
2. torch.nn:神经网络模块,包含了各种层和损失函数。
3. torch.optim:优化器,包含了各种优化算法。
4. torch.autograd:自动求导模块,可以自动计算梯度。
5. torch.utils.data:数据加载和预处理模块,包含了各种数据集和数据加载器。
6. torchvision:图像处理模块,包含了各种图像处理函数和数据集。
7. torch.functional:包含了各种常用的函数,如ReLU、sigmoid等。
8. torch.distributions:概率分布模块,包含了各种常用的概率分布函数。
9. torch.utils:包含了各种工具函数,如保存和加载模型、定时器等。
10. torch.cuda:CUDA加速模块,可以在GPU上运行张量和模型。
阅读全文