torch.fx.experimental.unification.unification_tools.get_in

torch.fx.experimental.unification.unification_tools.get_in(keys, coll, default=None, no_default=False) [source]

Возвращает coll[i0][i1]…[iX], где [i0, i1, …, iX]==keys.

Если coll[i0][i1]…[iX] не найдено, возвращает default, если не указано no_default; в этом случае возникает исключение KeyError или IndexError.

get_in — это обобщение operator.getitem для вложенных структур данных, таких как словари и списки.

>>> transaction = {
...     "name": "Alice",
...     "purchase": {"items": ["Apple", "Orange"], "costs": [0.50, 1.25]},
...     "credit card": "5555-1234-1234-1234",
... }
>>> get_in(["purchase", "items", 0], transaction)
'Apple'
>>> get_in(["name"], transaction)
'Alice'
>>> get_in(["purchase", "total"], transaction)
>>> get_in(["purchase", "items", "apple"], transaction)
>>> get_in(["purchase", "items", 10], transaction)
>>> get_in(["purchase", "total"], transaction, 0)
0
>>> get_in(["y"], {}, no_default=True)
Traceback (most recent call last):
    ...
KeyError: 'y'

См. также

itertoolz.get operator.getitem

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

object