HAMModel.py 文件源码

def getSequenceRealLength(sequences):
    '''
    ??sequences?????
    input:[a_size,b_size,c_size],?????????????b_size??????0??c_size?tensor????
    return?????b_size????????
    '''
    abs_sequneces = tf.abs(sequences)

    #??????max is 0
    abs_max_seq = tf.reduce_max(abs_sequneces,reduction_indices = 2)

    max_seq_sign = tf.sign(abs_max_seq)

    #????0????????????
    real_len = tf.reduce_sum(max_seq_sign,reduction_indices = 1)

    return tf.cast(real_len,tf.int32)