经过网上查阅资料以后,在这里记录一下multi_apply函数
def multi_apply(func,*args,**kwargs):
pfunc = partial(func,**kwargs) if kwargs else func
map_results = map(pfunc,*args)
return tuple(map(list,zip(*map_results)))
作用:将输入的函数func作用在iterable每个参数上,这里指的是args。
其中partial函数是讲kwargs以字典的形式传入func中,作为pfunc默认参数。
map函数是将args传入pfunc函数中,并进行计算。
pytorch 代码记录一:multi_apply
最新推荐文章于 2023-07-12 19:41:25 发布