快捷方式

torch.cuda.comm.broadcast

torch.cuda.comm.broadcast(tensor, devices=None, *, out=None)[源代码]

将张量广播到指定的 GPU 设备。

参数
  • tensor (Tensor) – 要广播的张量。可以位于 CPU 或 GPU 上。

  • devices (Iterable[torch.device, strint], 可选) – 要广播到的 GPU 设备的迭代器。

  • out (Sequence[Tensor], 可选, 仅限关键字) – 用于存储输出结果的 GPU 张量。

注意

必须指定 devicesout 中的一个。

返回值

  • 如果指定了 devices

    一个元组,包含 tensor 的副本,放置在 devices 上。

  • 如果指定了 out

    一个元组,包含 out 张量,每个张量都包含 tensor 的副本。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取初学者和高级开发人员的深入教程

查看教程

资源

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

查看资源