torch.broadcast_shapes

torch.broadcast_shapes(*shapes) → Size [исходный код]

Аналог broadcast_tensors(), но для форм.

Эквивалентно torch.broadcast_tensors(*map(torch.empty, shapes))[0].shape, но позволяет не создавать промежуточные тензоры. Это полезно для широковещательного распространения тензоров с одинаковой формой пакета, но разной формой крайних справа измерений, например для широковещательного распространения векторов средних значений вместе с ковариационными матрицами.

Пример:

>>> torch.broadcast_shapes((2,), (3, 1), (1, 1, 1))
torch.Size([1, 3, 2])

Параметры:

*shapes (torch.Size) – Формы тензоров.

Возвращает:

Форма, совместимая со всеми входными формами.

Тип возвращаемого значения:

shape (torch.Size)

Вызывает исключение:

RuntimeError – Если формы несовместимы.