打印pytorch每一层的输入输出

        v = F.conv2d(input, self.weight, self.bias, self.stride,self.padding, self.dilation, self.groups)

        v = v.detach()

        v = v.cpu().numpy().flatten()

        fp = open(‘conv2d.txt‘, ‘w‘)

        for d in v:

            fp.write(str(d)+‘\n‘)

        print(‘conv2d has Convert Completely!‘) 

        fp.close()  

        v = label.detach()

        v = v.cpu().numpy().flatten()

        fp = open(‘input.txt‘, ‘w‘)

        for d in v:

            fp.write(str(d)+‘\n‘)

        print(‘input has Convert Completely!‘) 

        fp.close()