Python中将函数作为列表元素

灵感来自阅读Fedformer的MOE模块源码

---

代码：

kernel_size = [12,34]
def time_list(param):
    return param
def fc(kernel_size):    
    return [time_list(kernel) for kernel in kernel_size]
# instance_fc = fc(kernel_size)
for func in fc(kernel_size):
    f_1 = func
    print(f_1)
            

解释：

for循环中的func就是一个函数元素，也就是一个time_list(kernel)，kernel_size有多少个元素，fc(kernel_size)就会产生多少个time_list(kernel)函数，这里的kernel就是kernel_size这个列表中的元素

这种把函数作为列表元素的方式还是第一次见，不过这种写代码的方式也挺优雅的，这种思维可以用在编写的模型中。