dim: 维度位置
dim
Broadcasting自动实现了若干unsqueeze和expand操作,以使两个Tensor的shape一致,从而完成某些操作(往往是加法)。
unsqueeze
expand