torch.fx.experimental.unification.unification_tools.merge_with

torch.fx.experimental.unification.unification_tools.merge_with(func, *dicts, **kwargs) [исходный код]

Объединяет словари и применяет функцию к объединённым значениям

Один и тот же ключ может встречаться в нескольких словарях, и все значения, сопоставленные с этим ключом, будут переданы функции в виде списка, например func([val1, val2, …]).

>>> merge_with(sum, {1: 1, 2: 2}, {1: 10, 2: 20})
{1: 11, 2: 22}

>>> merge_with(first, {1: 1, 2: 2}, {2: 20, 3: 30})  
{1: 1, 2: 2, 3: 30}

См. также

merge

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

object