unpooling_2d.py 文件源码

def backward_gpu(self, x, gy):
        gx = cuda.cupy.empty_like(x[0])
    cuda.elementwise(
            'raw T in, S indices',
            'T out',
            '''
           out = in[indices];

            ''', 'unpool')(gy[0].reduced_view(), self.indices, gx)
        return gx,