Component | 描述 |
torch | 类似 NumPy 的 Tensor 库,具有强大的 GPU 支持 |
torch.autograd | 基于 Tape 的主动微分库,支持 Torch 中全部可微分的张量操纵 |
torch.jit | 编译堆栈 (TorchScript),用于从 PyTorch 代码创建可序列化和可优化的模子 |
torch.nn | 与 autograd 深度集成的神经网络库,旨在实现最大的灵活性 |
torch.multiprocessing | Python 多处置惩罚,但具有跨历程的神奇 torch Tensors 内存共享功能。适用于数据加载和 Hogwild 练习 |
torch.ultis | DataLoader 和其他实用函数以方便使用 |
Data Tpye | dytpe | CPU Tensor | GPU Tensor |
32bit floating point | torch.float32/torch.float | torch.FloatTensor | Torch.cuda.FloatTensor |
64bit floating point | torch.float64/torch.double | torch.DoubleTensor | Torch.cuda.DoubleTensor |
8bit integer(unsigned) | torch.uint8 | torch.ByteTensor | Torch.cuda.ByteTensor |
8bit integer(signed) | torch.int8 | torch.CharTensor | torch.cuda.CharTensor |
16bit integer(signed) | torch.int16/torch.short | torch.ShortTensor | Torch.cuda.ShortTensor |
32bit integer(signed) | torch.int32/torch.int | torch.IntTensor | Torch.cuda.IntTensor |
64bit integer(signed) | torch.int64/torch.long | torch.LongTensor | torch.cuda.LongTensor |
欢迎光临 IT评测·应用市场-qidao123.com技术社区 (https://dis.qidao123.com/) | Powered by Discuz! X3.4 |