torch.broadcast_shapes

torch.broadcast_shapes(*shapes) → Size [source]

Аналогично 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 – Если формы несовместимы.