EqualityConstraint

class torch.fx.experimental.symbolic_shapes.EqualityConstraint(warn_only, source_pairs, derived_equalities, phantom_symbols, relaxed_sources) [исходный код]

Представляет и определяет различные виды ограничений равенства между источниками входных данных.

«Пара источников» — это пара источников входных данных для динамических размерностей, которые заданы равными. Мы представляем source_pairs в лесу структур данных union-find, чтобы эффективно проверять, равны ли транзитивно два таких источника.

«Производное равенство» связывает источник входных данных с выражением относительно корня. Корнем может быть другой источник входных данных, соответствующий некоторой динамической размерности, или фантомный символ, который напрямую не представляет ни одну динамическую размерность. Мы представляем derived_equalities, включающие источники входных данных, в транзитивно замкнутой карте, чтобы эффективно проверять, равен ли транзитивно источник входных данных заданному выражению относительно другого источника входных данных. (ПРИМЕЧАНИЕ: В отличие от этого, определить, равен ли транзитивно источник входных данных заданному выражению относительно фантомного символа, просто: такие выражения уже находятся в канонической форме, поэтому задача сводится к проверке равенства символьных выражений.)