torch.fx.experimental.graph_gradual_typechecker.linear_check

torch.fx.experimental.graph_gradual_typechecker.linear_check(tensor_type, module_instance) [исходный код]

Проверяет, удовлетворяет ли тип входного тензора условиям линейной операции, и возвращает тип выходных данных на основе входных и выходных признаков, заданных module_instance

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

TensorType