PyTorch 张量基础速记

记录 PyTorch 里张量（Tensor）最常用的创建、变形与广播操作，作为按项目长期累积的学习笔记，后续会持续追加。

创建张量

最常用的几种方式：

import torch

a = torch.zeros(2, 3)        # 全 0
b = torch.randn(2, 3)        # 标准正态分布
c = torch.arange(0, 6)       # 0..5
d = c.reshape(2, 3)          # 变形为 2×3

广播规则

两个张量做逐元素运算时，从尾部维度开始对齐：维度相等、或其中一方为 1 即可广播。

• 形状 (2, 3) 与 (3,) → 可广播为 (2, 3)
• 形状 (2, 1) 与 (1, 3) → 可广播为 (2, 3)

小贴士：不确定能否广播时，先用 torch.broadcast_shapes() 验证一下。