unpooling_2d.py 文件源码

def forward_gpu(self, x):
    shape = self.shape.tolist()
    y = cuda.cupy.zeros((shape[0], shape[1], shape[2], shape[3]), dtype=x[0].dtype)

    cuda.elementwise(
            'T in, S indices',
            'raw T out',
            '''
           out[indices] = in;

            ''', 'unpool')(x[0], self.indices, y.reduced_view())

    return y,