98c87b768221e7c2d7db33f0cefbc8ca6336e16a,my/tensorflow/general.py,,flatten,#Any#Any#,95

Before Change

    :param shape:
    :return:
    
    shape_rank = len(shape)
    assert shape_rank >= dim
    keep = dim - 1
    out = [reduce(mul, shape[:shape_rank-keep], 1)] + shape[shape_rank-keep:]
    return out

After Change

def flatten(tensor, keep):
    assert keep >= 1
    flat = tf.reshape(tensor, [-1] + tensor.get_shape().as_list()[-keep:])
    return flat

Instances