快捷方式

torch.cuda.comm.broadcast

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

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

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

  • devices (Iterable[torch.device, str or int], optional) – GPU 设备的迭代器,将在这些设备之间进行广播。

  • out (Sequence[Tensor], optional, keyword-only) – 用于存储输出结果的 GPU 张量序列。

注意

devicesout 中必须且只能指定一个。

返回

  • 如果指定了 devices

    返回一个元组,包含 tensor 的副本,这些副本位于 devices 上。

  • 如果指定了 out

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

文档

查阅 PyTorch 的完整开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源