chain()函数

定义：

chain函数是Python的itertools模块中的一个函数，它用于将多个可迭代对象（如列表、元组等）连接在一起，形成一个新的可迭代对象。这个新的可迭代对象可以逐个访问原始可迭代对象中的元素。

示例一

from itertools import chain
 
list1 = [1, 2, 3]
list2 = ['a', 'b', 'c']
list3 = [4, 5, 6]
 
# 使用chain函数将三个列表连接在一起
result = list(chain(list1, list2, list3))
print(result)  # 输出：[1, 2, 3, 'a', 'b', 'c', 4, 5, 6]

示例二

from itertools import chain 
# some consonants 
consonants =['d', 'f', 'k', 'l', 'n', 'p'] 
# some vowels 
vowels =['a', 'e', 'i', 'o', 'u'] 
# resultatnt list 
res = list(chain(consonants, vowels)) 
# sorting the list 
res.sort() 
print(res)

chain.from_iterable()函数

用于从单个可迭代项中链接项目，也就是说只能传入一个可迭代对象，并且迭代更彻底。

from itertools import chain 
  
li =['ABC', 'DEF', 'GHI', 'JKL'] 
# using chain-single iterable. 
res1 = list(chain(li)) 
res2 = list(chain.from_iterable(li)) 
  
print("using chain:", res1) 
print("using chain.from_iterable:", res2)

using chain: [‘ABC’, ‘DEF’, ‘GHI’, ‘JKL’]
using chain.from_iterable: [‘A’, ‘B’, ‘C’, ‘D’, ‘E’, ‘F’, ‘G’, ‘H’, ‘I’, ‘J’, ‘K’, ‘L’]

使用 chain() 的一个常见场景是当你想对不同的集合中所有元素一起执行某些操作的时候

比如：此时我想用优化器同时将多个生成器（genA2B, genB2A）同时优化，多个鉴别器（disGA, disGB, disLA, disLB）同时优化

# 优化器
gen_params = itertools.chain(self.genA2B.parameters(), self.genB2A.parameters())
self.G_optim = torch.optim.Adam(gen_params, lr=self.args.lr, betas=(0.5, 0.999), weight_decay=self.args.weight_decay)
disc_params = itertools.chain(self.disGA.parameters(), self.disGB.parameters(), self.disLA.parameters(), self.disLB.parameters())
self.D_optim = torch.optim.Adam(disc_params, lr=self.args.lr, betas=(0.5, 0.999), weight_decay=self.args.weight_decay)

#itertools.chain()接受一个或多个可迭代对象作为输入参数。然后创建一个迭代器，一次连续的返回每个
#可迭代对象中的元素。这种方式要比先将序列合并再迭代要高效
for x in a+b:   #这个是先合并再遍历，不够高效
    pass
for x in chain(a,b):   #better,更省内存
    pass