torch.cuda.comm.reduce_add

torch.cuda.comm.reduce_add(inputs, destination=None) [source]

Суммирует тензоры с нескольких GPU.

Все входные данные должны иметь совпадающие формы, типы и расположение. Результирующий тензор будет иметь такую же форму, тип и расположение.

Параметры

• inputs (Iterable[Tensor]) – итерируемый список тензоров для суммирования.
• destination (int, optional) – устройство, на котором будет размещён результат (по умолчанию: текущее устройство).

Возвращает

Тензор, содержащий поэлементную сумму всех входных данных, размещённый на destination устройстве.