torch.gathertorch.scattergather、scatter、reduce、all_reduce、all_gather、broadcast torch.gather按照指定的维度取数图中维度从1开始第一个例子,给定的维度是行,因此沿着行进行取值第二个例子,给定的维度是列,因此沿着列进行取值 torch.scatter gather、scatter、reduce、all_reduce、all_gather、broadcast