torch.fx.experimental.unification.unification_tools.itemfilter

torch.fx.experimental.unification.unification_tools.itemfilter(predicate, d, factory=<class 'dict'>) [исходный код]

Фильтрует элементы словаря по элементу

>>> def isvalid(item):
...     k, v = item
...     return k % 2 == 0 and v < 4

>>> d = {1: 2, 2: 3, 3: 4, 4: 5}
>>> itemfilter(isvalid, d)
{2: 3}

См. также

keyfilter valfilter itemmap

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

dict[_K, _V]