快捷方式

torch.cuda.comm.reduce_add

torch.cuda.comm.reduce_add(inputs, destination=None)[源代码][源代码]

对来自多个 GPU 的张量进行求和。

所有输入应具有匹配的形状、dtype 和布局。输出张量将具有相同的形状、dtype 和布局。

参数
  • inputs (Iterable[Tensor]) – 可迭代的待相加张量。

  • destination (int, optional) – 输出张量将放置的设备(默认值:当前设备)。

返回

一个张量,包含所有输入按元素相加的结果,放置在 destination 设备上。

文档

访问 PyTorch 全面的开发者文档

查看文档

教程

获取面向初学者和高级开发者的深度教程

查看教程

资源

查找开发资源并获得问题解答

查看资源