torch.fx.node.map_aggregate

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

Рекурсивно применяет fn к каждому объекту, входящему в arg.

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

Примечание

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

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

ArgumentT