【动手学深度学习】实用程序类Accumulator

这里定义一个实用程序类Accumulator，用于对多个变量进行累加。

源代码为：

class Accumulator:
    def __init__(self, n):
        self.data = [0, 0] * n
    
    def add(self, *args):
        self.data = [a + float(b) for a, b in zip(self.data, args)]

    def reset(self):
        self.data = [0, 0] * len(self.data)
        
    def __getitem__(self, idx):
        return self.data[idx]

对于第一个函数__init__(self, n)：首先在初始化的时候会根据传进来的n的大小来创建n个空间，且初始化全部为0.0。

第二个函数add(self, *args)：虽然*args代表这里可以传入任意个参数，但是因为要和初始化的个数相同不然要报错。for a,b in zip(self.data,args)是把原来类中对应位置的data和新传入的args做 a + float(b)加法操作然后重新赋给该位置的data。从而达到累加器的累加效果。

reset函数即重新设置空间大小并初始化。

__getitem__实现类似数组的取操作。

举例说明：

class Accumulator:
    def __init__(self, n):
        self.data = [0, 0] * n
    
    def add(self, *args):
        self.data = [a + float(b) for a, b in zip(self.data, args)]

    def reset(self):
        self.data = [0, 0] * len(self.data)
        
    def __getitem__(self, idx):
        return self.data[idx]


metric = Accumulator(3)
print(list(metric))
metric.add(3, 6, 9, 12)
print(list(metric))
metric.add(1, 1, 1, 1)
print(list(metric))

输出结果如下：

[0, 0, 0, 0, 0, 0]
[3.0, 6.0, 9.0, 12.0]
[4.0, 7.0, 10.0, 13.0]