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 张量序列。
注意
devices
和out
中必须且只能指定一个。- 返回
- 如果指定了
devices
, 返回一个元组,包含
tensor
的副本,这些副本位于devices
上。
- 如果指定了
- 如果指定了
out
, 返回一个元组,包含
out
中的张量,每个张量都包含tensor
的副本。
- 如果指定了