Функция torch.broadcast_shapes в PyTorch
Справочное руководство Pytorch torch
torch.broadcast_shapesЭто функция в PyTorch, используемая для вычисления результата бродкастинга нескольких форм. Она возвращает форму, полученную после бродкастинга входных форм, без фактического создания тензоров.
Определение функции
torch.broadcast_shapes(*shapes)
Пример использования
Пример
import torch
# Вычислить результат бродкастинга формы
result = torch.broadcast_shapes((3, 1), (1, 4), (3, 4))
print("(3,1) + (1,4) + (3,4) ->", result)
# Вывод: (3, 4)
# Несколько форм
result = torch.broadcast_shapes((5,), (1, 5), (3, 1, 5))
print("(5,) + (1,5) + (3,1,5) ->", result)
# Вывод: (3, 1, 5)
# Одна форма
result = torch.broadcast_shapes((2, 3))
print("(2,3) ->", result)
# Вывод: (2, 3)
# Сообщается об ошибке, когда широковещание невозможно
try:
result = torch.broadcast_shapes((3,), (4,))
except RuntimeError as e:
print("Ошибка широковещания:", e)
# Вычислить результат бродкастинга формы
result = torch.broadcast_shapes((3, 1), (1, 4), (3, 4))
print("(3,1) + (1,4) + (3,4) ->", result)
# Вывод: (3, 4)
# Несколько форм
result = torch.broadcast_shapes((5,), (1, 5), (3, 1, 5))
print("(5,) + (1,5) + (3,1,5) ->", result)
# Вывод: (3, 1, 5)
# Одна форма
result = torch.broadcast_shapes((2, 3))
print("(2,3) ->", result)
# Вывод: (2, 3)
# Сообщается об ошибке, когда широковещание невозможно
try:
result = torch.broadcast_shapes((3,), (4,))
except RuntimeError as e:
print("Ошибка широковещания:", e)
Другие расширения