日常实习面经

实现平均池化，窗口大小为k*k，stride为1

       arr=[[1,2,3,5],[4,5,6,7],[3,2,1,1],[2,1,2,1]]
        def averagePooling(arr,k):
            N=len(arr)
            tot=0
            ans=[]
            ks=k*k
            tot1=sum([arr[i][j] for i in range(k) for j in range(k)])
            print(tot1)
            #先算每一行的第一个窗口平均池化后的值，从第二行开始使用前一行的计算结果
            ans.append([tot1/ks])
            for j in range(N-k):
                tot1+=sum(arr[j+k][:k])
                tot1-=sum(arr[j][:k])
                ans.append([tot1/ks])
            #计算对应行每一列的窗口平均池化值，每次使用前一列的计算结果。第一列的结果已经通过上面的代码保存了
            for i in range(N-k+1):
                tot1=ans[i][0]*ks
                for j in range(N-k):
                    tot1+=sum(arr[z][j+k] for z in range(i,i+k))
                    tot1-=sum(arr[z][j] for z in range(i,i+k))
                    ans[i].append(tot1/ks)
            return ans
        print(averagePooling(arr,2))