ops.py 文件源码

def _fit(tensor, width):
    actual = tf.shape(tensor)[-1]
    result = tf.cond(tf.greater(actual, width),
                     lambda: _trim(tensor, width),
                     lambda: _pad(tensor, width))
    return result