- torch
- torch.Tensor
- Tensor Attributes
- torch.sparse
- torch.cuda
- torch.Storage
- torch.nn
- torch.nn.functional
- torch.autograd
- torch.optim
- torch.nn.init
- torch.distributions
- torch.multiprocessing
- torch.distributed
- torch.utils.bottleneck
- torch.utils.checkpoint
- torch.utils.cpp_extension
- torch.utils.data
- torch.utils.ffi
- torch.utils.model_zoo
- torch.onnx
- torch.legacy
包参考
torchvision
torchvision包朗阔了目前流行的数据集,模型结构和常用的图片转换工具。
如下代码用于获取加载图像的包的名称。
torchvision.get_image_backend()
指定用于加载图像的包。
torchvision.set_image_backend(backend)
参数:
backend(string)-backend代表图片名称。属于{'pil ','accimage'}之一。accimage包使用Intel IPP库。速度比 PIL快,但不支持多操作。
译者署名
| 用户名 | 头像 | 职能 | 签名 |
|---|---|---|---|
| Song | ![]() |
翻译 | 人生总要追求点什么 |
