torch.fx.experimental.unification.multipledispatch.utils.groupby#
- torch.fx.experimental.unification.multipledispatch.utils.groupby(func, seq)[source]#
Group a collection by a key function >>> names = [“Alice”, “Bob”, “Charlie”, “Dan”, “Edith”, “Frank”] >>> groupby(len, names) # doctest: +SKIP {3: [‘Bob’, ‘Dan’], 5: [‘Alice’, ‘Edith’, ‘Frank’], 7: [‘Charlie’]} >>> iseven = lambda x: x % 2 == 0 >>> groupby(iseven, [1, 2, 3, 4, 5, 6, 7, 8]) # doctest: +SKIP {False: [1, 3, 5, 7], True: [2, 4, 6, 8]} .. seealso::
countby
