torch.fx.experimental.unification.unification_tools.keymap

torch.fx.experimental.unification.unification_tools.keymap(func, d, factory=<class 'dict'>)

Apply function to keys of dictionary

>>> bills = {"Alice": [20, 15, 30], "Bob": [10, 35]}
>>> keymap(str.lower, bills)  
{'alice': [20, 15, 30], 'bob': [10, 35]}

See also

valmap itemmap

Return type:

dict[_K2, _V]