class Accumulator: # @save
"""在`n`个变量上累加。"""
def __init__(self, n):
self.data = [0.0] * n
print(self.data)
def add(self, *args): # *args可以接受不定长的参数,并打包成列表给args
for a, b in zip(self.data, args): # 分别输出一下a,b
print(a, b)
self.data = [a + float(b) for a, b in zip(self.data, args)]
print(self.data)
def reset(self):
self.data = [0.0] * len(self.data)
def __getitem__(self, idx):
return self.data[idx]
# 举例
metric = Accumulator(2) # [0.0] * 2
# [0.0, 0.0]
metric.add(4,5) # 传入打包成的[4,5],zip形成的映射对相加
# 0.0 4
# 0.0 5
# [4.0, 5.0]
李沐深度学习Accumulator函数
最新推荐文章于 2023-11-17 10:08:58 发布