PyTorch学习笔记(2)
# Broadcasting 自动扩展维度语法糖 # 关键点 A 和 B 张量做 broadcasting,如果 B 前面缺失一个维度,则自动在前面插入一个维度 插入的维度大小为 1,接下来会自动将其维度大小为 1 的维度扩张到和 A 张量相同位置维度一样的大小 例子:Bias [32, 1, 1] -> [1, 32, 1, 1] -> [4, 32, 14, 14] 所以 broadcasting 的本质就是 unsqueeze + expand # 案例 for actual demanding [class, students,...
more...







