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) – Формы тензоров.

Возвращает:

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

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

форма (torch.Size)

Возбуждает:

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