torch.cuda.comm.broadcast

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

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

Параметры

• tensor (Тензор) – тензор для трансляции. Может находиться на CPU или GPU.
• devices (Iterable[torch.device, строка или целое число], необязательно) – список устройств GPU, между которыми нужно выполнить трансляцию.
• out (Sequence[Тензор], необязательно, только ключевое слово) – список тензоров GPU, в которые будут сохранены результаты.

Примечание

Должен быть указан ровно один из devices и out.

Возвращает

• If devices is specified,

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

• If out is specified,

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