torch.fx.experimental.symbolic_shapes.compute_unbacked_bindings

torch.fx.experimental.symbolic_shapes.compute_unbacked_bindings(shape_env, example_value, old_example_value=None, peek=False) [исходный код]

После выполнения распространения фиктивных тензоров и получения результата example_value выполните обход example_value, чтобы найти вновь привязанные не подкреплённые символы и сохранить пути к ним для последующего использования. Если выделенный не подкреплённый SymInt не удаётся найти в example_value, возникает ошибка. (Примечание: если функция возвращает несколько результатов, необходимо вызвать эту функцию для кортежа выходных тензоров — ждать нельзя!)

Параметр peek позволяет проверить привязки, не изменяя затронутый список. Это главным образом полезно для проверки того, что real_tensor_prop_unbacked_vals своевременно заполняется при включённом propagate_real_tensors.

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

dict[Symbol, tuple[KeyEntry, …]] | None