torch.cuda.comm.broadcast

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

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

Параметры:

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

Примечание

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

Возвращает:

• If devices is specified,

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

• If out is specified,

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