47c6377324b4906ed5e133011350c22d1f809466,scripts/gan/srgan/train_srgan.py,SubpixelBlock,hybrid_forward,#SubpixelBlock#Any#Any#,37

Before Change

    def hybrid_forward(self, F, x,*args, **kwargs):
        x = self.conv(x)
        x = x.transpose([0, 2, 3, 1])
        batchsize,height,width,depth = x.shape
        x = x.reshape((batchsize, height , width, 2, 2, int(depth / 4)))
        x = x.transpose([0, 1,3,2,4,5])
        x = x.reshape((batchsize, height * 2, width * 2, int(depth / 4)))
        x = x.transpose([0, 3, 1, 2])
        x = self.relu(x)
        return x

class SRGenerator(gluon.nn.HybridBlock):

After Change

    def hybrid_forward(self, F, x, *args, **kwargs):
        x = self.conv(x)
        x = x.reshape((0, -4, -1, 4, 0, 0))  // bs, c // 4, h, w
        x = x.reshape((0, 0, -4, 2, 2, 0, 0))  // bs, c // 4, 2, 2, h, w
        x = x.transpose((0, 1, 2, 4, 3, 5))  // bs, c // 4, 2, h, 2, w
        x = x.reshape((0, 0, -3, -3))  // bs, c // 4, h * 2, w * 2
        x = self.relu(x)
        return x

Instances