torch.cuda.comm.broadcast

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

Рассылает тензор на указанные устройства GPU.

Параметры:

• tensor (Tensor) – тензор для рассылки. Может находиться на CPU или GPU.
• devices (Iterable[torch.device, str or int], optional) – итерируемый объект с устройствами GPU, между которыми выполняется рассылка.
• out (Sequence[Tensor], optional, keyword-only) – тензоры GPU для сохранения результатов.

Примечание

Необходимо указать ровно один из параметров devices и out.

Возвращает:

• If devices is specified,

  кортеж, содержащий копии tensor, размещённые на devices.

• If out is specified,

  кортеж, содержащий тензоры out, каждый из которых содержит копию tensor.