torch.fx.node.map_arg

torch.fx.node.map_arg(a, fn) [исходный код]

Рекурсивно применяет fn к каждому узлу Node, встречающемуся в arg.

arg может быть списком, кортежем, срезом или словарём со строковыми ключами: возвращаемое значение будет иметь тот же тип и структуру.

Примечание

Для этого API гарантируется обратная совместимость.

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

ArgumentT