读取数据（如mnist），格式.gz（源码）

#coding=utf-8

import os,gzip
import numpy as np
#    /home/wxy/Documents/tensorflow-generative-model-collections-master/data/mnist/train-labels-idx1-ubyte.gz
data_dir = '../tensorflow-generative-model-collections-master/data/mnist'

def extract_data(filename, num_data, head_size, data_size):
    with gzip.open(filename) as bytestream:
        data_head = bytestream.read(head_size)
        #前8/16位为包的信息（32位），数据的信息为整数（4位）：2049/2051(magic number)，60000(数据个数)，28（行数），28（列数）
        buf = bytestream.read(data_size * num_data)
        data = np.frombuffer(buf, dtype=np.uint8).astype(np.float)
        data_head1= np.frombuffer(data_head, dtype=np.uint8).astype(np.float)
    return data,data_head1

data,data_head1 = extract_data(data_dir + '/tr