torch.cuda.comm.reduce_add

torch.cuda.comm.reduce_add(inputs, destination=None) [исходный код]

Суммирует тензоры с нескольких графических процессоров.

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

Параметры:

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

Возвращает:

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