语言

PyTorch 快速参考

PyTorch 快速参考 的可打印速查表,适合快速查阅、学习和桌边参考

来源
PyTorch Documentation (pytorch.org)
许可证
MIT
版本
3 下载选项
更新时间
2026-03-27

示例

速查表包含什么

创建张量

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

下载

选择适合你阅读方式的版本

文本格式

以纯文本形式复制或下载

下载 Markdown

适合把命令、代码片段和配置复制到笔记、提示词或文档中

下载 Markdown