Откройте контекст
Выберите папку проекта и обозначьте нужные файлы или поведение.
Сценарии
Сценарии работы с кодом · Совместимость форм тензоров
03
Конкретная отправная точка
Определите общую форму тензоров по конечным измерениям с расширением единичных осей. Проверьте скалярные формы и пустые оси рядом с единичными, отклоняя несовместимые, отрицательные или дробные размеры.
Опишите изменение
Укажите входные данные, ожидаемый результат и границы, которые нужно сохранить.
Изучите работу
Проверьте предлагаемые изменения и команды, прежде чем полагаться на результат.
Проверьте результат
Сравните фактическое поведение с ожидаемым. Если проверка не проходит, разберитесь в причине перед следующими изменениями.
Совместимость форм тензоров
Конкретная отправная точка
from itertools import zip_longest
def broadcast_shape(a, b):
a, b = tuple(a), tuple(b)
if any(type(d) is not int or d < 0 for d in a + b):
raise ValueError("dimension.invalid")
result = []
for left, right in zip_longest(reversed(a), reversed(b), fillvalue=1):
if left == right or right == 1:
result.append(left)
elif left == 1:
result.append(right)
else:
raise ValueError("shape.incompatible")
return tuple(reversed(result))Пример — отправная точка для реального проекта, а не утверждение, что сгенерированный код уже протестирован. Адаптируйте пути, зависимости и ожидаемое поведение к своему приложению.