digit_binarized_quantized_weights.py 文件源码

def quantize_weight(W, precision=weight_quantization):
    '''
    For a given weight matrix, returns weights of values -1, 0 or 1
    :param W:
    :return:
    '''
    W_ = tf.round(W * precision) / precision
    return W_

########## Loading the dataset