GraphTypeChecker

class torch.fx.experimental.graph_gradual_typechecker.GraphTypeChecker(env, traced) [source]

type_check() [source]

Постепенная проверка типов графов. Эффект: после завершения проверки типов поле type каждого узла будет заполнено типом.

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

bool

type_check_node(n) [source]

Проверить тип заданного узла fx. Текущие операции: - Reshape - Transpose - Add - Relu - conv2d - batchnorm2d - flatten - maxpool2d - adaptiveavgpool2d - linear

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

Any