语言
PyTorch 快速参考
PyTorch 快速参考 的可打印速查表，适合快速查阅、学习和桌边参考
示例
速查表包含什么
- 张量
- 自动微分
- 神经网络
- 数据加载
- 训练循环
- 优化器
- 损失函数
- 保存与加载
- GPU
- 常用模式
创建张量
import torch
a = torch.tensor([1, 2, 3])
b = torch.zeros(2, 3)
c = torch.ones(3, 3)
d = torch.randn(2, 4) # 正态分布
张量构造函数
torch.zeros(m, n) | 全零张量，形状 (m, n) |
torch.ones(m, n) | 全一张量，形状 (m, n) |
torch.randn(m, n) | 标准正态分布随机张量 |
torch.arange(start, end, step) | 等差数列 |
torch.linspace(start, end, steps) | 固定数量的均匀点 |
torch.eye(n) | 单位矩阵 |
torch.empty(m, n) | 未初始化内存 |
与 NumPy 互操作
t = torch.from_numpy(np_array)
arr = tensor.numpy() # 共享内存
t = torch.as_tensor(np_array)
下载完整速查表，获取所有章节的打印排版版本
下载 PDF下载
